[XLA:GPU] add heuristic to collective permute decomposer and make it only decompose one CP
PiperOrigin-RevId: 731068852
Showing
- third_party/xla/xla/service/collective_permute_decomposer.cc 22 additions, 0 deletionsthird_party/xla/xla/service/collective_permute_decomposer.cc
- third_party/xla/xla/service/collective_permute_decomposer_test.cc 21 additions, 26 deletions...rty/xla/xla/service/collective_permute_decomposer_test.cc
Please register or sign in to comment