File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -442,7 +442,7 @@ def __init__(self):
442442 self .add_include ("vector" , "tuple" )
443443 self .add_include ("string" )
444444
445- if CUMM_CPU_ONLY_BUILD :
445+ if CUMM_CPU_ONLY_BUILD or compat . IsAppleSiliconMacOs :
446446 _STATIC_FUNC = pccm .static_function
447447 else :
448448 _STATIC_FUNC = pccm .cuda .static_function
@@ -451,7 +451,7 @@ def __init__(self):
451451 @_STATIC_FUNC
452452 def get_compiled_cuda_version (self ):
453453 code = pccm .code ()
454- if CUMM_CPU_ONLY_BUILD :
454+ if CUMM_CPU_ONLY_BUILD or compat . IsAppleSiliconMacOs :
455455 code .raw (f"return std::make_tuple(-1, -1);" )
456456 else :
457457 code .add_dependency (_CudaInclude )
@@ -590,6 +590,7 @@ def algo_can_be_nvrtc_compiled(self):
590590 code = pccm .code ()
591591 code .arg ("min_arch" , "std::tuple<int, int>" )
592592 cuda_ver_to_max_arch = [
593+ ((12 , 8 ), (10 , 0 )),
593594 ((11 , 8 ), (9 , 0 )),
594595 ((11 , 1 ), (8 , 6 )),
595596 ((11 , 0 ), (8 , 0 )),
You can’t perform that action at this time.
0 commit comments