Fix pre-compile on cpu-only machines (#7168)

+ Fix pre-compile on cpu-only machines

---------

Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
Co-authored-by: Olatunji Ruwase <tunji.ruwase@snowflake.com>
This commit is contained in:
Feng Yunlong
2025-08-12 22:24:30 +08:00
committed by GitHub
parent a54c394392
commit 8aadf6cbe4

View File

@ -245,6 +245,8 @@ class CUDA_Accelerator(DeepSpeedAccelerator):
return self._communication_backend_name
def is_triton_supported(self):
if not self.is_available():
return False
major, _ = torch.cuda.get_device_capability()
if major >= 8:
return True