Fix no-torch pipeline by not erroring out with cpu as new default acc…

…elerator
microsoft · Dec 18, 2024 · e5f0db2 · e5f0db2
1 parent 2f32966
commit e5f0db2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/.github/workflows/no-torch.yml b/.github/workflows/no-torch.yml
@@ -4,6 +4,7 @@ on:
   workflow_dispatch:
   pull_request:
     paths:
+      - 'accelerator/**'
       - '.github/workflows/no-torch.yml'
       - 'op_builder/**'
   schedule:

diff --git a/accelerator/cpu_accelerator.py b/accelerator/cpu_accelerator.py
@@ -3,9 +3,13 @@
 
 # DeepSpeed Team
 
-import torch
 from .abstract_accelerator import DeepSpeedAccelerator
 
+try:
+    import torch
+except ImportError as e:
+    pass
+
 try:
     import oneccl_bindings_for_pytorch  # noqa: F401 # type: ignore
     oneccl_imported_p = True