Adding raft distance specialization

rapidsai · Sep 30, 2022 · 25ebd2c · 25ebd2c
1 parent 2e98138
commit 25ebd2c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cpp/bench/spatial/knn.cu b/cpp/bench/spatial/knn.cu
@@ -19,6 +19,12 @@
 #include <raft/random/rng.cuh>
 
 #include <raft/spatial/knn/ivf_flat.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