Use specialization header in cagra test

rapidsai · Apr 18, 2023 · cb5b7d0 · cb5b7d0
1 parent 0754df5
commit cb5b7d0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cpp/test/neighbors/ann_cagra.cuh b/cpp/test/neighbors/ann_cagra.cuh
@@ -39,6 +39,12 @@
 #include <string>
 #include <vector>
 
+#ifdef RAFT_COMPILED
+#include <raft/neighbors/specializations/cagra.cuh>
+#else
+#pragma message("Specializations are not enabled; expect very long building times.")
+#endif
+
 namespace raft::neighbors::experimental::cagra {
 
 struct AnnCagraInputs {