EnzymeAD / EnzymeAD/Enzyme-JAX
Missing reduce constant prop
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
```
%61 = stablehlo.reduce(%c_0 init: %c_7) applies stablehlo.and across dimensions = [0, 1] : (tensor<1x140xi1>, tensor) -> tensor
```
https://github.com/openxla/stablehlo/blob/172164449ff2ddc8e5f6c12c29bc96eb68f30139/stablehlo/reference/Ops.cpp#L129
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by reading stablehlo/reference/Ops.cpp around line 129 to understand the referenced reduce behavior, then trace the corresponding constant-propagation path in Enzyme-JAX. Check existing tests for reduction folding and define done as correctly folding the shown stablehlo.and reduction, with regression coverage for this shape.
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
- Quiet
- Clarity
- Needs clarification
- Newbie friendliness
- 42/100