PR #17022: Add slicing -> reduce-scatter for dynamic-slice-fusion #17225
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
![copybara-service[bot]](https://avatars.githubusercontent.com/in/44061?s=60&v=4){kind=small}
copybara-service
bot
force-pushed
the
test_675256743
branch
13 times, most recently
from
c964afc
to
1ce947a
Compare
Imported from GitHub PR #17022 This patch adds support for slicing operations (`dynamic-slice` and `slice`) on operands of reduce-scatter in dynamic-slice-fusion. Copybara import of the project: -- 70d09c2 by Shraiysh Vaishay <[email protected]>: Add slicing -> reduce-scatter for dynamic-slice-fusion This patch adds support for slicing operations (dynamic-slice and slice) on operands of reduce-scatter in dynamic-slice-fusion. The command buffer also supports dynamic-slice-fusion where we only have static slices. So, these operations can now be a part of the cuda graph. Merging this change closes #17022 COPYBARA_INTEGRATE_REVIEW=#17022 from shraiysh:reduce_scatter_slicing_dus_fusion 70d09c2 PiperOrigin-RevId: 675871194
copybara-service
bot
force-pushed
the
test_675256743
branch
from
1ce947a
to
409ecf9
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR #17022: Add slicing -> reduce-scatter for dynamic-slice-fusion
Imported from GitHub PR #17022
This patch adds support for slicing operations (
dynamic-sliceandslice) on operands of reduce-scatter in dynamic-slice-fusion.Copybara import of the project:
--
70d09c2 by Shraiysh Vaishay [email protected]:
Add slicing -> reduce-scatter for dynamic-slice-fusion
This patch adds support for slicing operations (dynamic-slice and slice) on
operands of reduce-scatter in dynamic-slice-fusion.
The command buffer also supports dynamic-slice-fusion where we only have
static slices. So, these operations can now be a part of the cuda graph.
Merging this change closes #17022
FUTURE_COPYBARA_INTEGRATE_REVIEW=#17022 from shraiysh:reduce_scatter_slicing_dus_fusion 70d09c2