From 46c6c9e6ce9ede0eeb8a909958a4fecd2fbd942e Mon Sep 17 00:00:00 2001 From: Logan Adams <114770087+loadams@users.noreply.github.com> Date: Mon, 27 Jan 2025 16:15:47 -0800 Subject: [PATCH] Add CUDA 12.8 support and comment on CUDA 12.7 (#6975) --- op_builder/builder.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/op_builder/builder.py b/op_builder/builder.py index ab26054bda7d..cdd11f00cabf 100644 --- 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 }