Enzyme-JAX icon indicating copy to clipboard operation
Enzyme-JAX copied to clipboard

Published 20 hours ago • verified publisher icon EnzymeAD

Transpose Optimizations Missing (Move Transposes Up)

Open avik-pal opened this issue 7 months ago • 5 comments

test_sharded.mlir.txt

  • [x] transpose (slice x) -> slice (transpose ) (in https://github.com/EnzymeAD/Enzyme-JAX/pull/567)
  • [x] transpose reshape
  • [x] transpose (binary op) -- several of these will get fused with broadcast_in_dims
  • [x] transpose (concat)
  • [ ] reshape transpose

avik-pal avatar Mar 31 '25 02:03 avik-pal