EnzymeAD / EnzymeAD/Enzyme-JAX
Add an IR level option to disable auto-batching for a specific while loop
Open
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
For iterative algorithms, it might be worthwhile to avoid partial raising
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by clarifying what “partial raising” means and how auto-batching is represented for while loops in the Enzyme-JAX IR. The issue names no files, tests, or entry points, so identify the relevant IR option and define the expected behavior before implementation.
Written by the indexing model from the issue text.
Assessment
- Domain
- compilers
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100