Update mlu_accelerator.py

microsoft · Sep 13, 2024 · ee97f8d · ee97f8d
1 parent f7a7d09
commit ee97f8d
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/accelerator/mlu_accelerator.py b/accelerator/mlu_accelerator.py
@@ -10,11 +10,6 @@
 import torch
 # During setup stage torch may not be installed, pass on no torch will
 # allow op builder related API to be executed.
-try:
-    import torch_mlu
-except ImportError:
-    pass
-
 
 class MLU_Accelerator(DeepSpeedAccelerator):