Update distributed.py (#615)

# Pull Request

## What problem does this PR solve?

Issue Number: Fixed #

## Possible side effects?

- Performance:

- Backward compatibility:

Author: anakinxc

spu/utils/distributed.py

@@ -24,6 +24,8 @@
     set_framework,
     SAMPLE_NODES_DEF,
     SAMPLE_DEVICES_DEF,
+    dtype_spu_to_np,
+    shape_spu_to_np,
 )