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

2025-12-27 19:56:27 +08:00 · 2025-12-27 19:56:27 +08:00 · 06705fdcb3
parent a52dc60ba3
commit 06705fdcb3
1 changed files with 1 additions and 1 deletions
--- 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()