fix the accelerator issue

Co-authored-by: eigen2017 <[email protected]>
microsoft · Apr 12, 2024 · e448b6a · e448b6a
1 parent 0a654d6
commit e448b6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/accelerator/xpu_accelerator.py b/accelerator/xpu_accelerator.py
@@ -291,7 +291,7 @@ def export_envs(self):
         return []
 
     def visible_devices_envs(self):
-        return ['CUDA_VISIBLE_DEVICES']
+        return ['ZE_AFFINITY_MASK']
 
     def set_visible_devices_envs(self, current_env, local_accelerator_ids):
         for env in self.visible_devices_envs():