EnzymeAD / EnzymeAD/Enzyme-JAX
remove unnecessary intermediate insertdims
Open
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
```
%0 = stablehlo.broadcast_in_dim %arg1, dims = [0, 2, 4, 6] : (tensor<256x4x128x32xf32>) -> tensor<256x1x4x16x128x1x32x1xf32>
%1 = stablehlo.reshape %0 : (tensor<256x1x4x16x128x1x32x1xf32>) -> tensor<256x64x128x32xf32>
```
Contributor guide
No contributing guide indexed for this repository
Research direction
No source files or tests are named. Start by locating the rewrite or lowering logic that handles stablehlo.broadcast_in_dim followed by reshape, then reproduce the tensor pattern shown in the issue. Done means the unnecessary intermediate insertdims are removed without changing the resulting shape or behavior, with regression coverage if the relevant test area is found.
Written by the indexing model from the issue text.
Assessment
- Domain
- compilers
- Issue type
- Refactor
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 30/100