Skip to content

Commit 91a01c0

Browse files
committed
Update the valid compute capabilities.
1 parent 6ed05e6 commit 91a01c0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

kernel_tuner/utils/nvcuda.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
except ImportError:
88
cuda = None
99

10-
NVRTC_VALID_CC = np.array(["50", "52", "53", "60", "61", "62", "70", "72", "75", "80", "87", "89", "90", "90a"])
10+
NVRTC_VALID_CC = np.array(["50", "52", "53", "60", "61", "62", "70", "72", "75", "80", "87", "89", "90", "90a", "100", "100f", "100a", "101", "101f", "101a", "103", "103f", "103a", "120", "120f", "120a", "121", "121f", "121a"])
1111

1212

1313
def cuda_error_check(error):
@@ -28,4 +28,4 @@ def cuda_error_check(error):
2828

2929
def to_valid_nvrtc_gpu_arch_cc(compute_capability: str) -> str:
3030
"""Returns a valid Compute Capability for NVRTC `--gpu-architecture=`, as per https://docs.nvidia.com/cuda/nvrtc/index.html#group__options."""
31-
return max(NVRTC_VALID_CC[NVRTC_VALID_CC <= compute_capability], default="52")
31+
return max(NVRTC_VALID_CC[NVRTC_VALID_CC <= compute_capability], default="75")

0 commit comments

Comments
 (0)