Update minor CUDA version compatibility. (#4613)

Co-authored-by: Jeff Rasley <jerasley@microsoft.com>
microsoft · Nov 4, 2023 · 3437a5b · 3437a5b
1 parent ff53c22
commit 3437a5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/op_builder/builder.py b/op_builder/builder.py
@@ -70,7 +70,7 @@ def get_default_compute_capabilities():
         "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: ["12.0", "12.1", "12.2", "12.3"],
 }