ywt/fix: memory leak and precision error (#764)

* fix: memory leak and precision error * fix: mock is_initialized
DeepLink-org · Apr 7, 2024 · aea639a · aea639a
1 parent 8542a62
commit aea639a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/dipu/torch_dipu/dipu/__init__.py b/dipu/torch_dipu/dipu/__init__.py
@@ -29,8 +29,6 @@
 # as "gpu" or "cuda" (mock cuda is another problem)
 # only partially aligned now,
 __all__ = [
-    # resume initialize flag after random generator ready
-    # "is_initialized",
     # device
     "can_device_access_peer",
     "current_device",
@@ -42,6 +40,7 @@
     "get_device_properties",
     "get_device_capability",
     "is_available",
+    "is_initialized",
     "set_device",
     "GetDeviceProxy",
     "GetDeviceStaticProxy",