Making sure nn and dist libs are compiled when benchmarks are compiled

rapidsai · Oct 2, 2022 · 6471940 · 6471940
1 parent 61a0166
commit 6471940
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/build.sh b/build.sh
@@ -199,7 +199,9 @@ fi
 
 if hasArg bench || (( ${NUMARGS} == 0 )); then
     BUILD_BENCH=ON
+    COMPILE_DIST_LIBRARY=ON
     ENABLE_NN_DEPENDENCIES=ON
+    COMPILE_NN_LIBRARY=ON
     CMAKE_TARGET="${CMAKE_TARGET};bench_raft"
 fi