[WIP] [FEA] IVF_PQ build_from_args API

cpp/CMakeLists.txt

@@ -480,6 +480,7 @@ if(NOT BUILD_CPU_ONLY)
     src/neighbors/ivf_pq/detail/ivf_pq_build_extend_half_int64_t.cu
     src/neighbors/ivf_pq/detail/ivf_pq_build_extend_int8_t_int64_t.cu
     src/neighbors/ivf_pq/detail/ivf_pq_build_extend_uint8_t_int64_t.cu
+    src/neighbors/ivf_pq/detail/ivf_pq_build_precomputed_int64_t.cu
     src/neighbors/ivf_pq/detail/ivf_pq_compute_similarity_half_fp8_false.cu
     src/neighbors/ivf_pq/detail/ivf_pq_compute_similarity_half_fp8_true.cu
     src/neighbors/ivf_pq/detail/ivf_pq_compute_similarity_half_half.cu

cpp/src/neighbors/ivf_pq/detail/ivf_pq_build_precomputed_inst.cuh

@@ -0,0 +1,67 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+#pragma once
+
+#include <cuvs/neighbors/ivf_pq.hpp>
+
+#include "../ivf_pq_build.cuh"
+
+namespace cuvs::neighbors::ivf_pq {
+#define CUVS_INST_IVF_PQ_BUILD_PRECOMPUTED(IdxT)                                                   \
+  auto build(                                                                                      \
+    raft::resources const& handle,                                                                 \
+    const cuvs::neighbors::ivf_pq::index_params& index_params,                                     \
+    const uint32_t dim,                                                                            \
+    raft::host_mdspan<const float, raft::extent_3d<uint32_t>, raft::row_major> pq_centers,         \
+    raft::host_matrix_view<const float, uint32_t, raft::row_major> centers,                        \
+    std::optional<raft::host_matrix_view<const float, uint32_t, raft::row_major>> centers_rot,     \
+    std::optional<raft::host_matrix_view<const float, uint32_t, raft::row_major>> rotation_matrix) \
+    -> cuvs::neighbors::ivf_pq::index<IdxT>                                                        \
+  {                                                                                                \
+    return detail::build<IdxT>(                                                                    \
+      handle, index_params, dim, pq_centers, centers, centers_rot, rotation_matrix);               \
+  }                                                                                                \
+  void build(                                                                                      \
+    raft::resources const& handle,                                                                 \
+    const cuvs::neighbors::ivf_pq::index_params& index_params,                                     \
+    const uint32_t dim,                                                                            \
+    raft::host_mdspan<const float, raft::extent_3d<uint32_t>, raft::row_major> pq_centers,         \
+    raft::host_matrix_view<const float, uint32_t, raft::row_major> centers,                        \
+    std::optional<raft::host_matrix_view<const float, uint32_t, raft::row_major>> centers_rot,     \
+    std::optional<raft::host_matrix_view<const float, uint32_t, raft::row_major>> rotation_matrix, \
+    cuvs::neighbors::ivf_pq::index<IdxT>* idx)                                                     \
+  {                                                                                                \
+    detail::build<IdxT>(                                                                           \
+      handle, index_params, dim, pq_centers, centers, centers_rot, rotation_matrix, idx);          \
+  }                                                                                                \
+  auto build(                                                                                      \
+    raft::resources const& handle,                                                                 \
+    const cuvs::neighbors::ivf_pq::index_params& index_params,                                     \
+    const uint32_t dim,                                                                            \
+    raft::device_mdspan<const float, raft::extent_3d<uint32_t>, raft::row_major> pq_centers,       \
+    raft::device_matrix_view<const float, uint32_t, raft::row_major> centers,                      \
+    raft::device_matrix_view<const float, uint32_t, raft::row_major> centers_rot,                  \
+    raft::device_matrix_view<const float, uint32_t, raft::row_major> rotation_matrix)              \
+    -> cuvs::neighbors::ivf_pq::index<IdxT>                                                        \
+  {                                                                                                \
+    return detail::build<IdxT>(                                                                    \
+      handle, index_params, dim, pq_centers, centers, centers_rot, rotation_matrix);               \
+  }                                                                                                \
+  void build(                                                                                      \
+    raft::resources const& handle,                                                                 \
+    const cuvs::neighbors::ivf_pq::index_params& index_params,                                     \
+    const uint32_t dim,                                                                            \
+    raft::device_mdspan<const float, raft::extent_3d<uint32_t>, raft::row_major> pq_centers,       \
+    raft::device_matrix_view<const float, uint32_t, raft::row_major> centers,                      \
+    raft::device_matrix_view<const float, uint32_t, raft::row_major> centers_rot,                  \
+    raft::device_matrix_view<const float, uint32_t, raft::row_major> rotation_matrix,              \
+    cuvs::neighbors::ivf_pq::index<IdxT>* idx)                                                     \
+  {                                                                                                \
+    detail::build<IdxT>(                                                                           \
+      handle, index_params, dim, pq_centers, centers, centers_rot, rotation_matrix, idx);          \
+  }
+
+}  // namespace cuvs::neighbors::ivf_pq

cpp/src/neighbors/ivf_pq/detail/ivf_pq_build_precomputed_int64_t.cu

@@ -0,0 +1,15 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+#include <cuvs/neighbors/ivf_pq.hpp>
+
+#include "ivf_pq_build_precomputed_inst.cuh"
+
+namespace cuvs::neighbors::ivf_pq {
+CUVS_INST_IVF_PQ_BUILD_PRECOMPUTED(int64_t);
+
+#undef CUVS_INST_IVF_PQ_BUILD_PRECOMPUTED
+
+}  // namespace cuvs::neighbors::ivf_pq