From b41bff6b1bbf6520b7094daa2c76a1bc4270f449 Mon Sep 17 00:00:00 2001
From: Malte Foerster <mfoerster@nvidia.com>
Date: Mon, 2 Dec 2024 16:47:22 +0000
Subject: [PATCH] add fp16 header

---
 cpp/include/cuvs/neighbors/quantization.hpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cpp/include/cuvs/neighbors/quantization.hpp b/cpp/include/cuvs/neighbors/quantization.hpp
index fb087dcec..78c8c262f 100644
--- 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 {
 
 /**