Spaces:
Sleeping
Sleeping
Diego Devesa
commited on
Commit
·
22e1593
1
Parent(s):
0d836df
cuda : fix CUDA_FLAGS not being applied (llama/10403)
Browse files
ggml/src/ggml-blas/CMakeLists.txt
CHANGED
|
@@ -75,7 +75,6 @@ if (BLAS_FOUND)
|
|
| 75 |
|
| 76 |
message(STATUS "BLAS found, Includes: ${BLAS_INCLUDE_DIRS}")
|
| 77 |
|
| 78 |
-
#add_compile_options(${BLAS_LINKER_FLAGS})
|
| 79 |
target_compile_options(ggml-blas PRIVATE ${BLAS_LINKER_FLAGS})
|
| 80 |
|
| 81 |
if (${BLAS_INCLUDE_DIRS} MATCHES "mkl" AND (${GGML_BLAS_VENDOR} MATCHES "Generic" OR ${GGML_BLAS_VENDOR} MATCHES "Intel"))
|
|
|
|
| 75 |
|
| 76 |
message(STATUS "BLAS found, Includes: ${BLAS_INCLUDE_DIRS}")
|
| 77 |
|
|
|
|
| 78 |
target_compile_options(ggml-blas PRIVATE ${BLAS_LINKER_FLAGS})
|
| 79 |
|
| 80 |
if (${BLAS_INCLUDE_DIRS} MATCHES "mkl" AND (${GGML_BLAS_VENDOR} MATCHES "Generic" OR ${GGML_BLAS_VENDOR} MATCHES "Intel"))
|
ggml/src/ggml-cuda/ggml/CMakeLists.txt
CHANGED
|
@@ -149,7 +149,7 @@ if (CUDAToolkit_FOUND)
|
|
| 149 |
list(APPEND CUDA_FLAGS -Xcompiler ${CUDA_CXX_FLAGS_JOINED})
|
| 150 |
endif()
|
| 151 |
|
| 152 |
-
|
| 153 |
else()
|
| 154 |
message(FATAL_ERROR "CUDA Toolkit not found")
|
| 155 |
endif()
|
|
|
|
| 149 |
list(APPEND CUDA_FLAGS -Xcompiler ${CUDA_CXX_FLAGS_JOINED})
|
| 150 |
endif()
|
| 151 |
|
| 152 |
+
target_compile_options(ggml-cuda PRIVATE "$<$<COMPILE_LANGUAGE:CUDA>:${CUDA_FLAGS}>")
|
| 153 |
else()
|
| 154 |
message(FATAL_ERROR "CUDA Toolkit not found")
|
| 155 |
endif()
|