Merge branch 'loadams/fix-no-torch-failure-mlu' into loadams/fix-trig…

…gers-no-torch-workflow
deepspeedai · Sep 27, 2024 · a257e50 · a257e50
2 parents f57b2ae + 54ce016
commit a257e50
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/op_builder/mlu/fused_adam.py b/op_builder/mlu/fused_adam.py
@@ -5,7 +5,11 @@
 # DeepSpeed Team
 
 from .builder import MLUOpBuilder
-import torch
+
+try:
+    import torch
+except ImportError as e:
+    pass
 
 
 class MLUFusedAdam: