fix linux library dir

Pints-AI · Mar 24, 2024 · 0a13dfe · 0a13dfe
1 parent 7c28e0f
commit 0a13dfe
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -353,7 +353,12 @@ if (LLAMA_CUBLAS)
 
     find_package(CUDAToolkit)
     if (CUDAToolkit_FOUND)
-        link_directories(${FLASH_DIR}/build/Release)
+        if (WIN32)
+            link_directories(${FLASH_DIR}/build/Release)
+        else()
+            link_directories(${FLASH_DIR}/build)
+        endif()
+
         message(STATUS "cuBLAS found")
 
         enable_language(CUDA)