From 25ebd2c923d2bbfceaec145ff373ebc1400f5352 Mon Sep 17 00:00:00 2001
From: "Corey J. Nolet" <cjnolet@gmail.com>
Date: Fri, 30 Sep 2022 15:49:43 -0400
Subject: [PATCH] Adding raft distance specialization

---
 cpp/bench/spatial/knn.cu | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/cpp/bench/spatial/knn.cu b/cpp/bench/spatial/knn.cu
index 64a1217d7f..98721b4a63 100644
--- 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