EnzymeAD / EnzymeAD/Enzyme-JAX
Not removed reshape
Open
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
https://gist.github.com/wsmoses/9f398e6ac58a59c8615b0d76f34c2a08
becomes
https://gist.github.com/wsmoses/b03d64bf7c7ef9ed466656211e9e975c
cc @avik-pal @giordano
causes all-reduces here
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by comparing the two linked gists to identify the reshape operation that was not removed and the resulting all-reduces. Trace that behavior in the Enzyme-JAX compiler or differentiation pipeline, then verify that the reshape is removed and the unnecessary all-reduces no longer occur.
Written by the indexing model from the issue text.
Assessment
- Domain
- compilers
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100