Skip to content

[DO NOT MERGE] [FEA] KMeans - Batched (Streaming) Predict for Host Matrices#1962

Closed
tarang-jain wants to merge 2 commits intorapidsai:mainfrom
tarang-jain:batched-predict
Closed

[DO NOT MERGE] [FEA] KMeans - Batched (Streaming) Predict for Host Matrices#1962
tarang-jain wants to merge 2 commits intorapidsai:mainfrom
tarang-jain:batched-predict

Conversation

@tarang-jain
Copy link
Copy Markdown
Contributor

The predict function was separated out from #1886 into this PR. This PR is exploratory and we must evaluate the need for a host side predict function because the streaming of batches to device and calling of device side predict can be done by the user with the device side predict API itself.

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot bot commented Mar 28, 2026

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.

@aamijar aamijar added non-breaking Introduces a non-breaking change feature request New feature or request labels Apr 1, 2026
@tarang-jain
Copy link
Copy Markdown
Contributor Author

Closing this PR based on offline discussion. predict is trivially batch-able.

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

Labels

feature request New feature or request non-breaking Introduces a non-breaking change

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

2 participants