EnzymeAD / EnzymeAD/Enzyme-JAX
BatchOp(BroadcastInDim(), ....) -> Broadcast(BatchOp)
Open
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
If all operands fall under the following criteria:
- broadcast constructs dimensions that are batched. If we see this across all operands we should broadcast post batching.
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by locating the implementations of BatchOp, BroadcastInDim, and Broadcast, along with tests covering batching and broadcasting. Determine how batched dimensions are represented across all operands; done means the transformation broadcasts after batching when every operand meets the stated criteria, with tests demonstrating that behavior.
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
- 35/100