set reuse dist env

microsoft · Oct 12, 2024 · 4877418 · 4877418
1 parent e096310
commit 4877418
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tests/unit/common.py b/tests/unit/common.py
@@ -219,10 +219,13 @@ def _launch_daemonic_procs(self, num_procs):
         # Create process pool or use cached one
         master_port = None
 
-        if get_accelerator().device_name() == 'hpu':
-            if self.reuse_dist_env:
-                print("Ignoring reuse_dist_env for hpu")
-                self.reuse_dist_env = False
+        # if get_accelerator().device_name() == 'hpu':
+        #     if self.reuse_dist_env:
+        #         print("Ignoring reuse_dist_env for hpu")
+        #         self.reuse_dist_env = False
+
+        print("[MEM_DEBUG] Ignoring reuse_dist_env and forcibly setting it to False")
+        self.reuse_dist_env = False
 
         if self.reuse_dist_env:
             if num_procs not in self._pool_cache: