diff --git a/build.sh b/build.sh index 0caa823ca7..8ef2542faa 100755 --- a/build.sh +++ b/build.sh @@ -218,7 +218,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 diff --git a/cpp/bench/spatial/knn.cuh b/cpp/bench/spatial/knn.cuh index dedc9ea702..921932c791 100644 --- a/cpp/bench/spatial/knn.cuh +++ b/cpp/bench/spatial/knn.cuh @@ -23,6 +23,11 @@ #include #include #include + +#if defined RAFT_DISTANCE_COMPILED +#include +#endif + #if defined RAFT_NN_COMPILED #include #endif