From 5f409f6adb558ab0305ac58c2e30daa84fb63b5d Mon Sep 17 00:00:00 2001
From: "Corey J. Nolet" <cjnolet@gmail.com>
Date: Sat, 1 Oct 2022 22:41:43 -0400
Subject: [PATCH] Making proper changes

---
 build.sh                  | 2 ++
 cpp/bench/spatial/knn.cuh | 5 +++++
 2 files changed, 7 insertions(+)

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 <raft/spatial/knn/ivf_flat.cuh>
 #include <raft/spatial/knn/ivf_pq.cuh>
 #include <raft/spatial/knn/knn.cuh>
+
+#if defined RAFT_DISTANCE_COMPILED
+#include <raft/distance/specializations.cuh>
+#endif
+
 #if defined RAFT_NN_COMPILED
 #include <raft/spatial/knn/specializations.cuh>
 #endif