Refactor Vamana TUs to use vamana_matrix.json

cpp/CMakeLists.txt

@@ -1387,6 +1387,20 @@ if(NOT BUILD_CPU_ONLY)
     OUTPUT_FILE_FORMAT
       "${CMAKE_CURRENT_BINARY_DIR}/src/neighbors/nn_descent_inst_data_@data_abbrev@_index_@index_abbrev@.cu"
   )
+  generate_inst_matrix(
+    vamana_build_inst_files
+    MATRIX_JSON_FILE "${CMAKE_CURRENT_SOURCE_DIR}/src/neighbors/vamana_matrix.json"
+    INPUT_FILE "${CMAKE_CURRENT_SOURCE_DIR}/src/neighbors/vamana_build_inst.cu.in"
+    OUTPUT_FILE_FORMAT
+      "${CMAKE_CURRENT_BINARY_DIR}/src/neighbors/vamana_build_inst_data_@data_abbrev@.cu"
+  )
+  generate_inst_matrix(
+    vamana_serialize_inst_files
+    MATRIX_JSON_FILE "${CMAKE_CURRENT_SOURCE_DIR}/src/neighbors/vamana_matrix.json"
+    INPUT_FILE "${CMAKE_CURRENT_SOURCE_DIR}/src/neighbors/vamana_serialize_inst.cu.in"
+    OUTPUT_FILE_FORMAT
+      "${CMAKE_CURRENT_BINARY_DIR}/src/neighbors/vamana_serialize_inst_data_@data_abbrev@.cu"
+  )
   generate_inst_matrix(
     refine_device_inst_files
     MATRIX_JSON_FILE
@@ -1529,15 +1543,9 @@ if(NOT BUILD_CPU_ONLY)
     src/neighbors/sample_filter.cu
     src/neighbors/tiered_index.cu
     src/neighbors/sparse_brute_force.cu
-    src/neighbors/vamana_build_float.cu
-    src/neighbors/vamana_build_half.cu
-    src/neighbors/vamana_build_uint8.cu
-    src/neighbors/vamana_build_int8.cu
-    src/neighbors/vamana_codebooks_float.cu
-    src/neighbors/vamana_serialize_float.cu
-    src/neighbors/vamana_serialize_half.cu
-    src/neighbors/vamana_serialize_uint8.cu
-    src/neighbors/vamana_serialize_int8.cu
+    ${vamana_build_inst_files}
+    src/neighbors/vamana_codebooks.cu
+    ${vamana_serialize_inst_files}
     src/preprocessing/quantize/scalar.cu
     src/preprocessing/quantize/binary.cu
     src/preprocessing/quantize/pq.cu

cpp/src/neighbors/vamana.cuh

@@ -80,9 +80,12 @@ index<T, IdxT> build(
 template <typename T, typename IdxT>
 void serialize(raft::resources const& res,
                const std::string& file_prefix,
-               const index<T, IdxT>& index_)
+               const index<T, IdxT>& index_,
+               bool include_dataset = true,
+               bool sector_aligned  = false)
 {
-  cuvs::neighbors::vamana::detail::serialize<T, IdxT>(res, file_prefix, index_);
+  cuvs::neighbors::vamana::detail::serialize<T, IdxT>(
+    res, file_prefix, index_, include_dataset, sector_aligned);
 }
 
 template <typename T>

cpp/src/neighbors/vamana_build_inst.cu.in

@@ -0,0 +1,36 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+#include <cuvs/core/cuda_fp16.hpp>
+
+#include <cuvs/neighbors/vamana.hpp>
+#include <neighbors/vamana.cuh>
+
+namespace {
+
+using data_t  = @data_type@;
+using index_t = uint32_t;
+
+}  // namespace
+
+namespace cuvs::neighbors::vamana {
+
+auto build(raft::resources const& handle,
+           const index_params& params,
+           raft::device_matrix_view<const data_t, int64_t, raft::row_major> dataset)
+  -> index<data_t, index_t>
+{
+  return cuvs::neighbors::vamana::build<data_t, index_t>(handle, params, dataset);
+}
+
+auto build(raft::resources const& handle,
+           const index_params& params,
+           raft::host_matrix_view<const data_t, int64_t, raft::row_major> dataset)
+  -> index<data_t, index_t>
+{
+  return cuvs::neighbors::vamana::build<data_t, index_t>(handle, params, dataset);
+}
+
+}  // namespace cuvs::neighbors::vamana

cpp/src/neighbors/vamana_matrix.json

@@ -0,0 +1,20 @@
+{
+  "_data": [
+    {
+      "data_type": "float",
+      "data_abbrev": "f"
+    },
+    {
+      "data_type": "half",
+      "data_abbrev": "h"
+    },
+    {
+      "data_type": "int8_t",
+      "data_abbrev": "i8"
+    },
+    {
+      "data_type": "uint8_t",
+      "data_abbrev": "u8"
+    }
+  ]
+}

cpp/src/neighbors/vamana_serialize_inst.cu.in

@@ -0,0 +1,30 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+#include <cuvs/core/cuda_fp16.hpp>
+
+#include <cuvs/neighbors/vamana.hpp>
+#include <neighbors/vamana.cuh>
+
+namespace {
+
+using data_t  = @data_type@;
+using index_t = uint32_t;
+
+}  // namespace
+
+namespace cuvs::neighbors::vamana {
+
+void serialize(raft::resources const& handle,
+               const std::string& file_prefix,
+               const index<data_t, index_t>& index_,
+               bool include_dataset,
+               bool sector_aligned)
+{
+  cuvs::neighbors::vamana::serialize<data_t, index_t>(
+    handle, file_prefix, index_, include_dataset, sector_aligned);
+}
+
+}  // namespace cuvs::neighbors::vamana