Skip to content

Conversation

@eternalNight
Copy link
Contributor

PR #6993 replaces the flat IPG buffers with a dict maintaining type-indexed buckets. The member is also renamed from _ipg_bucket_flat_buffer to ipg_buckets.

Update the bucket clearing logic in init_z3 accordingly.

PR deepspeedai#6993 replaces the flat IPG buffers with a dict maintaining
type-indexed buckets. The member is also renamed from
`_ipg_bucket_flat_buffer` to `ipg_buckets`.

Update the bucket clearing logic in `init_z3` accordingly.

Signed-off-by: Junjie Mao <[email protected]>
Copy link
Collaborator

@tohtana tohtana left a comment

Choose a reason for hiding this comment

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

Thank you for the fix!

@tohtana tohtana enabled auto-merge (squash) October 1, 2025 03:20
@tohtana tohtana merged commit aa90f54 into deepspeedai:master Oct 1, 2025
12 checks passed
delock pushed a commit that referenced this pull request Oct 3, 2025
PR #6993 replaces the flat IPG buffers with a dict maintaining
type-indexed buckets. The member is also renamed from
`_ipg_bucket_flat_buffer` to `ipg_buckets`.

Update the bucket clearing logic in `init_z3` accordingly.

Signed-off-by: Junjie Mao <[email protected]>
Signed-off-by: Guokai Ma <[email protected]>
mauryaavinash95 pushed a commit to DataStates/DeepSpeed that referenced this pull request Oct 4, 2025
PR deepspeedai#6993 replaces the flat IPG buffers with a dict maintaining
type-indexed buckets. The member is also renamed from
`_ipg_bucket_flat_buffer` to `ipg_buckets`.

Update the bucket clearing logic in `init_z3` accordingly.

Signed-off-by: Junjie Mao <[email protected]>
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.

2 participants