Просмотр исходного кода

cuda : fix CUDA_FLAGS not being applied (#10403)

Diego Devesa 1 год назад
Родитель
Сommit
3ee6382d48
2 измененных файлов с 1 добавлено и 2 удалено
  1. 0 1
      ggml/src/ggml-blas/CMakeLists.txt
  2. 1 1
      ggml/src/ggml-cuda/CMakeLists.txt

+ 0 - 1
ggml/src/ggml-blas/CMakeLists.txt

@@ -75,7 +75,6 @@ if (BLAS_FOUND)
 
     message(STATUS "BLAS found, Includes: ${BLAS_INCLUDE_DIRS}")
 
-    #add_compile_options(${BLAS_LINKER_FLAGS})
     target_compile_options(ggml-blas PRIVATE ${BLAS_LINKER_FLAGS})
 
     if (${BLAS_INCLUDE_DIRS} MATCHES "mkl" AND (${GGML_BLAS_VENDOR} MATCHES "Generic" OR ${GGML_BLAS_VENDOR} MATCHES "Intel"))

+ 1 - 1
ggml/src/ggml-cuda/CMakeLists.txt

@@ -149,7 +149,7 @@ if (CUDAToolkit_FOUND)
         list(APPEND CUDA_FLAGS -Xcompiler ${CUDA_CXX_FLAGS_JOINED})
     endif()
 
-    add_compile_options("$<$<COMPILE_LANGUAGE:CUDA>:${CUDA_FLAGS}>")
+    target_compile_options(ggml-cuda PRIVATE "$<$<COMPILE_LANGUAGE:CUDA>:${CUDA_FLAGS}>")
 else()
     message(FATAL_ERROR "CUDA Toolkit not found")
 endif()