|
5 | 5 |
|
6 | 6 | #pragma once |
7 | 7 |
|
8 | | -#include "common.hpp" |
9 | 8 | #include <cuda_fp16.h> |
10 | 9 | #include <cuvs/distance/distance.hpp> |
11 | 10 | #include <cuvs/neighbors/common.hpp> |
@@ -558,7 +557,7 @@ struct index : cuvs::neighbors::index { |
558 | 557 | * `detail::clone_any_dataset_view_for_cagra_index` stores a shallow copy of the view variant. |
559 | 558 | * Supported: |
560 | 559 | * `empty_dataset_view`, `indirect_dataset_view`, `device_padded_dataset_view`, |
561 | | - * `dataset_view<strided_dataset_container,...>`. The index stores only a **non-owning** view; the |
| 560 | + * `strided_dataset_view<T, int64_t>`. The index stores only a **non-owning** view; the |
562 | 561 | * caller must keep all underlying device storage (and any `indirect_dataset_view` target) alive |
563 | 562 | * for the index lifetime. |
564 | 563 | * |
@@ -693,8 +692,7 @@ struct index : cuvs::neighbors::index { |
693 | 692 | static_cast<unsigned>(required_stride), |
694 | 693 | static_cast<unsigned>(src_stride)); |
695 | 694 |
|
696 | | - ::cuvs::neighbors::dataset_view<strided_dataset_container, T, dataset_index_type, true, false> |
697 | | - wrap(dataset_view); |
| 695 | + ::cuvs::neighbors::strided_dataset_view<T, dataset_index_type> wrap(dataset_view); |
698 | 696 | update_dataset(res, cuvs::neighbors::any_dataset_view<T, dataset_index_type>(wrap)); |
699 | 697 | } |
700 | 698 |
|
@@ -1420,7 +1418,7 @@ auto build_ace(raft::resources const& res, |
1420 | 1418 | * is used, returns `build_result` with `.vpq` that the caller must keep alive. |
1421 | 1419 | * See `build(res, params, device_matrix_view)` for full documentation. |
1422 | 1420 | * |
1423 | | - * Strided device rows (`dataset_view<strided_dataset_container, T, int64_t, true, false>`) are |
| 1421 | + * Strided device rows (`strided_dataset_view<T, int64_t>`) are |
1424 | 1422 | * carried as the strided alternative inside `any_dataset_view` and convert implicitly from that |
1425 | 1423 | * view type (`any_dataset_view<T, int64_t>(strided_view)` is optional). |
1426 | 1424 | */ |
|
0 commit comments