Skip to content

Commit 46c6c9e

Browse files
authored
Add CUDA 12.8 support and comment on CUDA 12.7 (#6975)
1 parent 1640f6d commit 46c6c9e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

op_builder/builder.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,8 @@ def get_default_compute_capabilities():
7676
cuda_minor_mismatch_ok = {
7777
10: ["10.0", "10.1", "10.2"],
7878
11: ["11.0", "11.1", "11.2", "11.3", "11.4", "11.5", "11.6", "11.7", "11.8"],
79-
12: ["12.0", "12.1", "12.2", "12.3", "12.4", "12.5", "12.6"],
79+
12: ["12.0", "12.1", "12.2", "12.3", "12.4", "12.5", "12.6",
80+
"12.8"], # There does not appear to be a CUDA Toolkit 12.7
8081
}
8182

8283

0 commit comments

Comments
 (0)