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

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