Add CUDA 12.8 support and comment on CUDA 12.7 (#6975)

deepspeedai · Jan 28, 2025 · 46c6c9e · 46c6c9e
1 parent 1640f6d
commit 46c6c9e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/op_builder/builder.py b/op_builder/builder.py
@@ -76,7 +76,8 @@ def get_default_compute_capabilities():
 cuda_minor_mismatch_ok = {
     10: ["10.0", "10.1", "10.2"],
     11: ["11.0", "11.1", "11.2", "11.3", "11.4", "11.5", "11.6", "11.7", "11.8"],
-    12: ["12.0", "12.1", "12.2", "12.3", "12.4", "12.5", "12.6"],
+    12: ["12.0", "12.1", "12.2", "12.3", "12.4", "12.5", "12.6",
+         "12.8"],  # There does not appear to be a CUDA Toolkit 12.7
 }