|
|
@@ -61,7 +61,7 @@ if (CUDAToolkit_FOUND)
|
|
|
set(CMAKE_CUDA_ARCHITECTURES ${PROCESSED_ARCHITECTURES})
|
|
|
else()
|
|
|
foreach(ARCH ${CMAKE_CUDA_ARCHITECTURES})
|
|
|
- if(ARCH MATCHES "^12[0-9]$")
|
|
|
+ if(ARCH MATCHES "^12[0-9](-real|-virtual)?$")
|
|
|
message(FATAL_ERROR "Compute capability ${ARCH} used, use ${ARCH}a or ${ARCH}f for Blackwell specific optimizations")
|
|
|
endif()
|
|
|
endforeach()
|