EnzymeAD / EnzymeAD/Reactant.jl
should `fold` and similar operations be unrolled?
Open
- Dominant language
- Julia
- Stars
- 370
- Forks
- 74
- Avg merge
- 18h 47m
- Merged PRs (30d)
- 30
Description
(needs https://github.com/LuxDL/Lux.jl/pull/1026 for the dump)
```julia
using Lux, Reactant, Random
model = Recurrence(RNNCell(4 => 4))
ps, st = Lux.setup(Xoshiro(123), model) |> Reactant.to_rarray
x = rand(Float32, 4, 2, 12) |> Reactant.ConcreteRArray
@code_hlo model(x, ps, st)
```
This currently gets unrolled into the following monstrosity: https://pastebin.com/QE00SCqb
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.