Merge branch 'duli/capability' of https://github.com/microsoft/DeepSpeed

into duli/capability
microsoft · Apr 22, 2024 · 6ad67c7 · 6ad67c7
2 parents 159c8e4 + 7588dc6
commit 6ad67c7
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/accelerator/abstract_accelerator.py b/accelerator/abstract_accelerator.py
@@ -291,6 +291,14 @@ def build_extension(self):
     def export_envs(self):
         ...
 
+    @abc.abstractmethod
+    def visible_devices_envs(self):
+        ...
+
+    @abc.abstractmethod
+    def set_visible_devices_envs(self, current_env, local_accelerator_ids):
+        ...
+
     def get_ds_feature(self, key):
         return self._ds_features[key]