Skip to content

Commit 4dedaf4

Browse files
fix mac build
1 parent 8d08fde commit 4dedaf4

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

cumm/common.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff 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)),

0 commit comments

Comments
 (0)