add fp16 header

rapidsai · Dec 2, 2024 · b41bff6 · b41bff6
1 parent 7dc93f0
commit b41bff6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cpp/include/cuvs/neighbors/quantization.hpp b/cpp/include/cuvs/neighbors/quantization.hpp
@@ -23,6 +23,8 @@
 #include <raft/core/host_mdarray.hpp>
 #include <raft/core/host_mdspan.hpp>
 
+#include <cuda_fp16.h>
+
 namespace cuvs::neighbors::quantization {
 
 /**