Skip to content
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.

Sign up for GitHub

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

Jump to bottom

[NVIDIA GPU] [XLA_GPU_MS_COLLECTIVE] Add utility functions for async collective stream assignment #23522

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

[NVIDIA GPU] [XLA_GPU_MS_COLLECTIVE] Add utility functions for async collective stream assignment #23522

wants to merge 2 commits into from
+208 −0

Conversation

terryysun
Copy link
Contributor

@terryysun terryysun commented Mar 8, 2025 •
edited
Loading

Breaking #22450 into 3 steps:

  1. Utility functions with unit tests.
  2. Update to execution stream assignment.
  3. Runtime integration.

This is the first step where we introduce the util functions needed and test them throughly.

@terryysun terryysun requested a review from frgossen March 8, 2025 00:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@frgossen frgossen Awaiting requested review from frgossen

At least 1 approving review is required to merge this pull request.

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@terryysun