EnzymeAD / EnzymeAD/Enzyme-JAX
Add support to raise an affine.for to stablehlo.while
Open
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
To currently support affine.for, there is a PR which does an unrolling pass https://github.com/EnzymeAD/Enzyme-JAX/pull/441. This is a temporary hack to lower a for op but long term, we should raise this to stablehlo.while with support for induction variables.
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.