fix the accelerator issue

Co-authored-by: eigen2017 <[email protected]>
microsoft · Apr 15, 2024 · c89600c · c89600c
1 parent e448b6a
commit c89600c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/accelerator/mps_accelerator.py b/accelerator/mps_accelerator.py
@@ -259,7 +259,8 @@ def build_extension(self):
     def export_envs(self):
         return []
 
-    def visible_devices_env(self):
+    def visible_devices_envs(self):
+        # TODO could not find visible devices env for mps
         return ['CUDA_VISIBLE_DEVICES']
 
     def set_visible_devices_envs(self, current_env, local_accelerator_ids):