Sfoglia il codice sorgente

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

Aman Gupta 1 mese fa
parent
commit
06705fdcb3
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      ggml/src/ggml-cuda/CMakeLists.txt

+ 1 - 1
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()