Skip to content

Expose pq list data to python#1428

Merged
rapids-bot[bot] merged 11 commits intorapidsai:mainfrom
benfred:pq_list_python
Jan 6, 2026
Merged

Expose pq list data to python#1428
rapids-bot[bot] merged 11 commits intorapidsai:mainfrom
benfred:pq_list_python

Conversation

@benfred
Copy link
Copy Markdown
Contributor

@benfred benfred commented Oct 14, 2025

Expose pq list data to python. Also change cuvsMatrixCopy to handle more than just 2D tensors - so that things like pq_centers can be converting to host memory without needing to involve cupy .

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented Oct 14, 2025

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@benfred benfred added improvement Improves an existing functionality non-breaking Introduces a non-breaking change labels Oct 25, 2025
@benfred benfred self-assigned this Oct 25, 2025
@benfred benfred marked this pull request as ready for review October 25, 2025 00:10
@benfred benfred requested review from a team as code owners October 25, 2025 00:10
Comment thread c/src/core/c_api.cpp Outdated
Comment thread c/src/core/c_api.cpp
} else {
// Otherwise use cudaMemcpyAsync - and assert that we don't have strided data
RAFT_EXPECTS(src.strides == nullptr, "cuvsCopyMatrix only supports strides with 2D inputs");
RAFT_EXPECTS(dst.strides == nullptr, "cuvsCopyMatrix only supports strides with 2D inputs");
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can this raft::copy be used to suppport strides in non-2D output?

@benfred benfred requested a review from a team as a code owner November 20, 2025 05:54
@cjnolet
Copy link
Copy Markdown
Member

cjnolet commented Jan 6, 2026

/merge

@rapids-bot rapids-bot Bot merged commit 7634551 into rapidsai:main Jan 6, 2026
93 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

improvement Improves an existing functionality non-breaking Introduces a non-breaking change

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

3 participants