cmake cuda: fix
This commit is contained in:
parent
5bf2b94dd4
commit
ce4a979c90
1 changed files with 3 additions and 2 deletions
|
@ -767,9 +767,10 @@ endif()
|
||||||
set(CUDA_CXX_FLAGS "")
|
set(CUDA_CXX_FLAGS "")
|
||||||
|
|
||||||
if (LLAMA_CUBLAS)
|
if (LLAMA_CUBLAS)
|
||||||
set(CUDA_FLAGS ${CXX_FLAGS} -use_fast_math)
|
set(CUDA_FLAGS -use_fast_math)
|
||||||
|
list(APPEND CUDA_CXX_FLAGS ${CXX_FLAGS})
|
||||||
if (NOT MSVC)
|
if (NOT MSVC)
|
||||||
list(APPEND CUDA_FLAGS -Wno-pedantic)
|
list(APPEND CUDA_CXX_FLAGS -Wno-pedantic)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
if (LLAMA_ALL_WARNINGS AND NOT MSVC)
|
if (LLAMA_ALL_WARNINGS AND NOT MSVC)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue