EnzymeAD / EnzymeAD/Enzyme-JAX
`whileOp` reduction into `reduce`
Open
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
```julia
s = ...
for i in 1:size(x, 2)
s += x[:, i, :]
end
return s
```
into
```julia
return reduce(+, x; dims=2)
```
We already have the unrolled version. This one is much easier to lift with the loop fission pass
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.