From 1e37d84e3a32e19fa6eee084c1c8e85227f2b03b Mon Sep 17 00:00:00 2001 From: Johnny Date: Wed, 20 Aug 2025 15:29:31 +0700 Subject: [PATCH] Merge pull request #27537 from johnnynunez:patch-2 Refactor Blackwell #27537 In CUDA 13: - 10.0 is b100/b200 same for aarch64 (gb200) - 10.3 is GB300 - 11.0 is Thor with new OpenRm driver (moves to SBSA) - 12.0 is RTX/RTX PRO - 12.1 is Spark GB10 Thor was moved from 10.1 to 11.0 and Spark is 12.1. Related patch: https://github.com/pytorch/pytorch/pull/156176 --- cmake/OpenCVDetectCUDAUtils.cmake | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/cmake/OpenCVDetectCUDAUtils.cmake b/cmake/OpenCVDetectCUDAUtils.cmake index 7f0f923380..b993be8894 100644 --- a/cmake/OpenCVDetectCUDAUtils.cmake +++ b/cmake/OpenCVDetectCUDAUtils.cmake @@ -109,7 +109,7 @@ macro(ocv_initialize_nvidia_device_generations) set(_arch_ampere "8.0;8.6") set(_arch_lovelace "8.9") set(_arch_hopper "9.0") - set(_arch_blackwell "10.0;12.0") + set(_arch_blackwell "10.0;10.3;11.0;12.0;12.1") if(NOT CMAKE_CROSSCOMPILING) list(APPEND _generations "Auto") endif() @@ -273,14 +273,15 @@ macro(ocv_set_cuda_arch_bin_and_ptx nvcc_executable) endif() if(NOT _nvcc_res EQUAL 0) message(STATUS "CUDA: Automatic detection of CUDA generation failed. Going to build for all known architectures") - # TX1 (5.3) TX2 (6.2) Xavier (7.2) V100 (7.0) Orin (8.7) Thor (10.1) + # TX1 (5.3) TX2 (6.2) Xavier (7.2) V100 (7.0) Orin (8.7) Thor (11.0) Spark (12.1) ocv_filter_available_architecture(${nvcc_executable} __cuda_arch_bin 5.3 6.2 7.2 7.0 8.7 - 10.1 + 11.0 + 12.1 ) else() set(__cuda_arch_bin "${_nvcc_out}")