diff --git a/op_builder/builder.py b/op_builder/builder.py index 0b4018e69616..aef7d50358dd 100644 --- a/op_builder/builder.py +++ b/op_builder/builder.py @@ -83,8 +83,7 @@ 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.8"], # There does not appear to be a CUDA Toolkit 12.7 + 12: ["12.0", "12.1", "12.2", "12.3", "12.4", "12.5", "12.6", "12.8", "12.9"], # There is no CUDATk 12.7 }