Skip to content
Snippets Groups Projects
Commit b1d4ad64 authored by Philipp Hack's avatar Philipp Hack Committed by TensorFlower Gardener
Browse files

PR #10811: Restore ReshapeDecomposer And LayoutNormalization Passes in GPU Compiler 

Imported from GitHub PR https://github.com/openxla/xla/pull/10811

Restores individual ReshapeDecomposer and LayoutNormalization passes in the GPU compiler previously removed in #9852 and fixes failures in cudnn_norm_rewriter_test.cc.
Copybara import of the project:

--
52065712a6bd4f6bb874575d82b4d99b8f2d16dd by Philipp Hack <phack@nvidia.com>:

Restore ReshapeDecomposer and LayoutNormalization passes.

Merging this change closes #10811

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/10811 from philipphack:u_layer_reshape_decomposer_xla 52065712a6bd4f6bb874575d82b4d99b8f2d16dd
PiperOrigin-RevId: 617990157
parent f9f12ddc
Hide whitespace changes
Inline Side-by-side
Showing
with 86 additions and 7 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment