Exclude the other index includes from dynamic_batching.hpp

rapidsai · Dec 3, 2024 · 07ed312 · 07ed312
1 parent acec170
commit 07ed312
Show file tree

Hide file tree

Showing 7 changed files with 10 additions and 1,062 deletions.
diff --git a/cpp/include/cuvs/neighbors/dynamic_batching.hpp b/cpp/include/cuvs/neighbors/dynamic_batching.hpp
@@ -18,10 +18,6 @@
 
 #include <cuvs/neighbors/common.hpp>
 
-#include <cuvs/neighbors/cagra.hpp>
-#include <cuvs/neighbors/ivf_flat.hpp>
-#include <cuvs/neighbors/ivf_pq.hpp>
-
 namespace cuvs::neighbors::dynamic_batching {
 
 namespace detail {

diff --git a/cpp/src/neighbors/detail/dynamic_batching.cuh b/cpp/src/neighbors/detail/dynamic_batching.cuh
@@ -37,6 +37,7 @@
 #include <chrono>
 #include <limits>
 #include <memory>
+#include <variant>
 #include <vector>
 
 #ifndef CUVS_SYSTEM_LITTLE_ENDIAN