From 1077710c5aa2759c6ad2375097f9734f50db6e0c Mon Sep 17 00:00:00 2001
From: anakinxc <103552181+anakinxc@users.noreply.github.com>
Date: Thu, 21 Mar 2024 15:27:47 +0800
Subject: [PATCH] Update distributed.py (#617)

# Pull Request

## What problem does this PR solve?

Issue Number: Fixed #

## Possible side effects?

- Performance:

- Backward compatibility:
---
 spu/utils/distributed.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/spu/utils/distributed.py b/spu/utils/distributed.py
index 051dff80..8cf56399 100644
--- a/spu/utils/distributed.py
+++ b/spu/utils/distributed.py
@@ -26,6 +26,7 @@
     SAMPLE_DEVICES_DEF,
     dtype_spu_to_np,
     shape_spu_to_np,
+    save,
 )