Update accelerator/mps_accelerator.py

microsoft · Apr 20, 2024 · b4d465b · b4d465b
1 parent d75c634
commit b4d465b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/accelerator/mps_accelerator.py b/accelerator/mps_accelerator.py
@@ -259,6 +259,7 @@ def build_extension(self):
     def export_envs(self):
         return []
 
+    # TODO: mpu's visible envs is confirmed, keep as CUDA_VISIBLE_DEVICES
     def visible_devices_envs(self):
         # TODO: could not find visible devices env for mps
         return ['CUDA_VISIBLE_DEVICES']