mirror of
https://github.com/deepspeedai/DeepSpeed.git
synced 2025-10-20 15:33:51 +08:00
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:
@ -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
|
||||
|
Reference in New Issue
Block a user