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

PR #18152: [XLA:GPU] Avoid fusion-wrapping copies #18192

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

copybara-service[bot]
Copy link

PR #18152: [XLA:GPU] Avoid fusion-wrapping copies

Imported from GitHub PR #18152

Fusion wrapping copies breaks the logic for detecting copies from copy-insertion in rematerialization pass.

This patch avoids wrapping copy instructions and instead emits them directly in IrEmitterUnnested.

This should fix #17922
Copybara import of the project:

--
49daad1 by Jaroslav Sevcik [email protected]:

Avoid fusion-wrapping copies

Merging this change closes #18152

FUTURE_COPYBARA_INTEGRATE_REVIEW=#18152 from jaro-sevcik:avoid-fusion-wrapping-copies 49daad1

Imported from GitHub PR #18152

Fusion wrapping copies breaks the logic for detecting copies from copy-insertion in rematerialization pass.

This patch avoids wrapping copy instructions and instead emits them directly in IrEmitterUnnested.

This should fix #17922
Copybara import of the project:

--
49daad1 by Jaroslav Sevcik <[email protected]>:

Avoid fusion-wrapping copies

Merging this change closes #18152

FUTURE_COPYBARA_INTEGRATE_REVIEW=#18152 from jaro-sevcik:avoid-fusion-wrapping-copies 49daad1
PiperOrigin-RevId: 684709374
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.

[Regression] Gradient explodes after upgrading to JAX 0.4.33 from 0.4.30
1 participant