Skip to content

Reuse raft::mdarray/mdspan directly in dense dataset storage - #2530

Open
HowardHuang1 wants to merge 1 commit into
NVIDIA:mainfrom
HowardHuang1:hh-abstract-common-dataset-functions-one-level-up-and-reuse-mdarray-26_10
Open

Reuse raft::mdarray/mdspan directly in dense dataset storage#2530
HowardHuang1 wants to merge 1 commit into
NVIDIA:mainfrom
HowardHuang1:hh-abstract-common-dataset-functions-one-level-up-and-reuse-mdarray-26_10

Conversation

@HowardHuang1

@HowardHuang1 HowardHuang1 commented Aug 30, 2026

Copy link
Copy Markdown
Contributor

Addresses #2395.

dense_owning_matrix/dense_view_matrix (and the VPQ codebook/code matrix aliases) picked between raft::device_matrix and raft::host_matrix via std::conditional_t, even though those are themselves just aliases for raft::mdarray/raft::mdspan with the exact accessor already computed as Accessor. Point the aliases at raft::mdarray/raft::mdspan directly instead.

dense_row_major_dataset_owning_storage/_view_storage also wrapped their matrix/view as a field and hand-forwarded view()/data_handle(), which raft::mdarray/raft::mdspan already provide natively. They now inherit from the matrix/view type instead, so those forwards are no longer needed.

…2395)

dense_owning_matrix/dense_view_matrix (and the VPQ codebook/code matrix
aliases) picked between raft::device_matrix and raft::host_matrix via
std::conditional_t, even though those are themselves just aliases for
raft::mdarray/raft::mdspan with the exact accessor already computed as
Accessor. Point the aliases at raft::mdarray/raft::mdspan directly instead.

dense_row_major_dataset_owning_storage/_view_storage also wrapped their
matrix/view as a field and hand-forwarded view()/data_handle(), which
raft::mdarray/raft::mdspan already provide natively. They now inherit from
the matrix/view type instead, so those forwards are no longer needed.

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
@HowardHuang1
HowardHuang1 requested a review from a team as a code owner August 30, 2026 23:52
@copy-pr-bot

copy-pr-bot Bot commented Aug 30, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant