Skip to content

[WIP] Support strided matrix view as an input to matrix::samples_rows #2626

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 6 commits into
base: branch-25.06
Choose a base branch
from

Conversation

enp1s0
Copy link
Member

@enp1s0 enp1s0 commented Apr 3, 2025

This PR adds support for strided_matrix_view as an input to matrix::sample_rows(). For that, it also adds raft::make_host_strided_matrix_view and the support for strided_matrix_view as an input to matrix::gather().

Rel: rapidsai/cuvs#716 (comment)

todo

  • Update gather
  • Update sample_rows
  • Update gather test
  • Update sample_rows test

@enp1s0 enp1s0 requested a review from a team as a code owner April 3, 2025 06:31
@enp1s0 enp1s0 self-assigned this Apr 3, 2025
@github-actions github-actions bot added the cpp label Apr 3, 2025
@enp1s0 enp1s0 changed the title [WIP][FEA] Support strided matrix view as an input to matrix::samples_rows [WIP] Support strided matrix view as an input to matrix::samples_rows Apr 3, 2025
@enp1s0 enp1s0 added feature request New feature or request non-breaking Non-breaking change labels Apr 3, 2025
@cjnolet
Copy link
Member

cjnolet commented Apr 25, 2025

/ok to test b709a7f

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cpp feature request New feature or request non-breaking Non-breaking change
Projects
Status: In Progress
Development

Successfully merging this pull request may close these issues.

2 participants