EnzymeAD / EnzymeAD/Enzyme-JAX
Check if dimension is dynamic and use dynamic_reshape instead of reshape in affinetostablehlo raising
Open
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
https://github.com/EnzymeAD/Enzyme-JAX/blob/main/src/enzyme_ad/jax/Passes/AffineToStableHLORaising.cpp
see an example of dynamic reshape here: https://github.com/EnzymeAD/Enzyme-JAX/pull/1677
Contributor guide
No contributing guide indexed for this repository
Research direction
Open src/enzyme_ad/jax/Passes/AffineToStableHLORaising.cpp and compare its reshape handling with the dynamic reshape example in pull request 1677. Done means dynamic dimensions use dynamic_reshape while the existing reshape path remains appropriate for static dimensions.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp
- Domain
- compilers
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Clearly specified
- Newbie friendliness
- 55/100