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.

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

[XLA:LatencyHidingScheduler] Let GetResourcesFromInstruction return a complete list of resources used by instructions in a while loop. This will make async done and while ops have similar priority (in terms of occupying resource types) and avoid delaying the while loops only because they cross the overlap limit (even though they have a higher async depth). #22119

Merged
merged 1 commit into from
Feb 10, 2025

Conversation

copybara-service[bot]
Copy link

@copybara-service copybara-service bot commented Jan 30, 2025

[XLA:LatencyHidingScheduler] Let GetResourcesFromInstruction return a complete list of resources used by instructions in a while loop. This will make async done and while ops have similar priority (in terms of occupying resource types) and avoid delaying the while loops only because they cross the overlap limit (even though they have a higher async depth).

This CL also fixes the double counting of a resource in GetNumResourcesPerInstruction because of multiple async done ops in the while body.

@copybara-service copybara-service bot changed the title [XLA:LatencyHidingScheduler] Let GetResourcesFromInstruction return a complete list of resources used by instructions in a while loop. This will make async done and while ops have similar priority (in terms of occupying resource types) and avoid delaying the while loops only because they cross the overlap limit. [XLA:LatencyHidingScheduler] Let GetResourcesFromInstruction return a complete list of resources used by instructions in a while loop. This will make async done and while ops have similar priority (in terms of occupying resource types) and avoid delaying the while loops only because they cross the overlap limit (even though they have a higher async depth). Feb 7, 2025
@copybara-service copybara-service bot force-pushed the test_721121051 branch 7 times, most recently from 1c11488 to 5d58fd6 Compare February 10, 2025 23:21
… a complete list of resources used by instructions in a while loop. This will make async `done` and while ops have similar priority (in terms of occupying resource types) and avoid delaying the while loops only because they cross the overlap limit (even though they have a higher async depth).

This CL also fixes the double counting of a resource in `GetNumResourcesPerInstruction` because of multiple async `done` ops in the while body.

PiperOrigin-RevId: 725375955
@copybara-service copybara-service bot merged commit 7f161ac into main Feb 10, 2025
@copybara-service copybara-service bot deleted the test_721121051 branch February 10, 2025 23:51
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.

1 participant