Skip to content

Commit 50e41b0

Browse files
johnnynunezD00E
authored andcommitted
Merge pull request opencv#27537 from johnnynunez:patch-2
Refactor Blackwell opencv#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: pytorch/pytorch#156176
1 parent a57401d commit 50e41b0

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

cmake/OpenCVDetectCUDAUtils.cmake

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ macro(ocv_initialize_nvidia_device_generations)
109109
set(_arch_ampere "8.0;8.6")
110110
set(_arch_lovelace "8.9")
111111
set(_arch_hopper "9.0")
112-
set(_arch_blackwell "10.0;12.0")
112+
set(_arch_blackwell "10.0;10.3;11.0;12.0;12.1")
113113
if(NOT CMAKE_CROSSCOMPILING)
114114
list(APPEND _generations "Auto")
115115
endif()
@@ -273,14 +273,15 @@ macro(ocv_set_cuda_arch_bin_and_ptx nvcc_executable)
273273
endif()
274274
if(NOT _nvcc_res EQUAL 0)
275275
message(STATUS "CUDA: Automatic detection of CUDA generation failed. Going to build for all known architectures")
276-
# TX1 (5.3) TX2 (6.2) Xavier (7.2) V100 (7.0) Orin (8.7) Thor (10.1)
276+
# TX1 (5.3) TX2 (6.2) Xavier (7.2) V100 (7.0) Orin (8.7) Thor (11.0) Spark (12.1)
277277
ocv_filter_available_architecture(${nvcc_executable} __cuda_arch_bin
278278
5.3
279279
6.2
280280
7.2
281281
7.0
282282
8.7
283-
10.1
283+
11.0
284+
12.1
284285
)
285286
else()
286287
set(__cuda_arch_bin "${_nvcc_out}")

0 commit comments

Comments
 (0)