EnzymeAD / EnzymeAD/Enzyme-JAX

`whileOp` reduction into `reduce`

Open
#1,506 1 comment 0 reactions 0 assignees View on GitHub
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.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.