ggml-cuda: Use same regex for GGML_NATIVE=OFF (llama/18407)

diff --git a/ggml/src/ggml-cuda/CMakeLists.txt b/ggml/src/ggml-cuda/CMakeLists.txt
index c0f8bcaa378f3975f9635cb1200f7036e8c5ca12..3b438c30ce64b6928960293d3adc8de3ca981798 100644 (file)
--- a/ggml/src/ggml-cuda/CMakeLists.txt
+++ b/ggml/src/ggml-cuda/CMakeLists.txt
@@ -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()