CHecking cudatoolkit library dir

rapidsai · Oct 3, 2023 · fce179b · fce179b
1 parent d539316
commit fce179b
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/cpp/bench/ann/CMakeLists.txt b/cpp/bench/ann/CMakeLists.txt
@@ -249,7 +249,8 @@ if(TARGET faiss::faiss_avx2)
 endif()
 
 message("RAFT_FAISS_TARGETS: ${RAFT_FAISS_TARGETS}")
-
+message(
+  "CUDAToolkit_LIBRARY_DIR: ${CUDAToolkit_LIBRARY_DIR})
 if(RAFT_ANN_BENCH_USE_FAISS_CPU_FLAT)
   ConfigureAnnBench(
     NAME FAISS_CPU_FLAT PATH bench/ann/src/faiss/faiss_cpu_benchmark.cpp LINKS
@@ -339,16 +340,17 @@ if(RAFT_ANN_BENCH_SINGLE_EXE)
   target_compile_definitions(
     ANN_BENCH
     PRIVATE
-      $<$<BOOL:${CUDAToolkit_FOUND}>:ANN_BENCH_LINK_CUDART="libcudart.so.${CUDAToolkit_VERSION_MAJOR}.${CUDAToolkit_VERSION_MINOR}.${CUDAToolkit_VERSION_PATCH}">
-      $<$<BOOL:${NVTX3_HEADERS_FOUND}>:ANN_BENCH_NVTX3_HEADERS_FOUND>
-  )
+      $<$<BOOL:${CUDAToolkit_FOUND}>:ANN_BENCH_LINK_CUDART="libcudart.so.${CUDAToolkit_VERSION_MAJOR}.${CUDAToolkit_VERSION_MINOR}.${CUDAToolkit_VERSION_PATCH}
+  ">
+  $<$<BOOL:${NVTX3_HEADERS_FOUND}>:ANN_BENCH_NVTX3_HEADERS_FOUND>
+)
 
-  target_link_options(ANN_BENCH PRIVATE -export-dynamic)
+target_link_options(ANN_BENCH PRIVATE -export-dynamic)
 
-  install(
-    TARGETS ANN_BENCH
-    COMPONENT ann_bench
-    DESTINATION bin/ann
-    EXCLUDE_FROM_ALL
-  )
+install(
+  TARGETS ANN_BENCH
+  COMPONENT ann_bench
+  DESTINATION bin/ann
+  EXCLUDE_FROM_ALL
+)
 endif()