Skip to content

[KMeans] Rename streaming_batch_size param#2328

Open
tarang-jain wants to merge 4 commits into
NVIDIA:release/26.08from
tarang-jain:rename-params
Open

[KMeans] Rename streaming_batch_size param#2328
tarang-jain wants to merge 4 commits into
NVIDIA:release/26.08from
tarang-jain:rename-params

Conversation

@tarang-jain

@tarang-jain tarang-jain commented Jul 20, 2026

Copy link
Copy Markdown
Contributor

This PR renames the streaming_batch_size param for kmeans to:

  1. Better convey the meaning of the param -- link to discussion: Exposing opt-in double buffering for MG batched KMeans #2323 (comment)
  2. Avoid confusion with the already existing batch_samples and batch_centroids params

The idea is that the streaming word can be misleading. We are "buffering up" a batch of samples onto the device.

Merge after #2329

@tarang-jain tarang-jain self-assigned this Jul 20, 2026
@tarang-jain tarang-jain added breaking Introduces a breaking change improvement Improves an existing functionality labels Jul 20, 2026
@copy-pr-bot

copy-pr-bot Bot commented Jul 20, 2026

Copy link
Copy Markdown

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.

@tarang-jain
tarang-jain changed the base branch from main to release/26.08 July 20, 2026 17:32
@tarang-jain
tarang-jain marked this pull request as ready for review July 20, 2026 17:46
@tarang-jain
tarang-jain requested review from a team as code owners July 20, 2026 17:46
@tarang-jain
tarang-jain requested a review from a team as a code owner July 20, 2026 19:06
@tarang-jain
tarang-jain requested a review from bdice July 20, 2026 19:06
@tarang-jain
tarang-jain removed request for a team and bdice July 21, 2026 03:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

breaking Introduces a breaking change improvement Improves an existing functionality

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant