Skip to content

Commit

Permalink
Fix ci: windows cublas
Browse files Browse the repository at this point in the history
Signed-off-by: Molly Sophia <[email protected]>
  • Loading branch information
MollySophia committed Jul 3, 2024
1 parent 970a813 commit c5df6ba
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -105,6 +105,7 @@ endif()

if (RWKV_CUBLAS)
cmake_minimum_required(VERSION 3.17)
set(CMAKE_CUDA_COMPILER_FORCED TRUE)

find_package(CUDAToolkit)

Expand Down Expand Up @@ -417,6 +418,11 @@ target_compile_features(ggml PUBLIC c_std_11) # Don't bump

if (MSVC)
target_link_libraries(ggml PUBLIC ${RWKV_EXTRA_LIBS} Threads::Threads)
if (RWKV_CUBLAS)
target_compile_options(ggml PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:
-allow-unsupported-compiler
>)
endif()
else()
if (WIN32 AND RWKV_HIPBLAS)
target_link_libraries(ggml PUBLIC ${RWKV_EXTRA_LIBS} Threads::Threads)
Expand Down

0 comments on commit c5df6ba

Please sign in to comment.