EnzymeAD / EnzymeAD/Enzyme-JAX

Missing reduce constant prop

Open
#1,084 3 comments 0 reactions 0 assignees View on GitHub
good first issue
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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.