Update distributed.py (#617)

# Pull Request ## What problem does this PR solve? Issue Number: Fixed # ## Possible side effects? - Performance: - Backward compatibility:
secretflow · Mar 21, 2024 · 1077710 · 1077710
1 parent 5c8c051
commit 1077710
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,
 )