EnzymeAD / EnzymeAD/Enzyme-JAX
Transpose Optimizations Missing (Move Transposes Up)
Open
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
[test_sharded.mlir.txt](https://github.com/user-attachments/files/19528762/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
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.