Add clarifying comment from cuda_accelerator

microsoft · Dec 18, 2024 · cfb86cc · cfb86cc
1 parent e5f0db2
commit cfb86cc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/accelerator/cpu_accelerator.py b/accelerator/cpu_accelerator.py
@@ -5,6 +5,8 @@
 
 from .abstract_accelerator import DeepSpeedAccelerator
 
+# During setup stage torch may not be installed, pass on no torch will
+# allow op builder related API to be executed.
 try:
     import torch
 except ImportError as e: