fix arch

FindDefinition · FindDefinition · commit 4c77b38d1ab5 · 2025-05-28T02:09:21.000Z
diff --git a/cumm/common.py b/cumm/common.py
@@ -91,7 +91,7 @@ def _get_cuda_arch_flags(is_gemm: bool = False) -> Tuple[List[str], List[Tuple[i
 
     supported_arches = [
         '3.5', '3.7', '5.0', '5.2', '6.0', '6.1', '7.0', '7.2', '7.5', '8.0',
-        '8.6', '8.9', '9.0'
+        '8.6', '8.9', '9.0', "10.0", "12.0",
     ]
     supported_arches += ['5.3', '6.2', '7.2', '8.7']
     valid_arch_strings = supported_arches + [

Original file line number	Diff line number	Diff line change
`@@ -91,7 +91,7 @@ def _get_cuda_arch_flags(is_gemm: bool = False) -> Tuple[List[str], List[Tuple[i`
`91`	`91`
`92`	`92`	`supported_arches = [`
`93`	`93`	`'3.5', '3.7', '5.0', '5.2', '6.0', '6.1', '7.0', '7.2', '7.5', '8.0',`
`94`		`- '8.6', '8.9', '9.0'`
	`94`	`+ '8.6', '8.9', '9.0', "10.0", "12.0",`
`95`	`95`	`]`
`96`	`96`	`supported_arches += ['5.3', '6.2', '7.2', '8.7']`
`97`	`97`	`valid_arch_strings = supported_arches + [`