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

This commit is contained in:
Logan Adams 2025-01-27 16:15:47 -08:00 коммит произвёл GitHub
Родитель 1640f6df4f
Коммит 46c6c9e6ce
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
1 изменённых файлов: 2 добавлений и 1 удалений

Просмотреть файл

@ -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
}