ScaledPlan: region subfield missing (needed for AD)
Open
- Dominant language
- Julia
- Stars
- 300
- Forks
- 61
- PR merge metrics
- No merged PRs in 30d
Description
The `ScaledPlan` like e.g. the result of `plan_fft` has no field `region`, leading to an error when attempting to AD it via Zygote. Zygote (which depends on AbstractFFT) is requiring a `region` subfield.
Example
```Julia
begin
using Flux
using Zygote
using LinearAlgebra
using FFTW
end
N = 15
data = rand(Float64, N)
FT = plan_fft(data) # e.g. plan_dct / plan_idct would work
iFT= plan_ifft(data)
m = Chain(x->FT*x, Dense(N,N), x->iFT*x)
loss(x,y) = sum(abs2,m(x) .- y)
loss(data,data) # precompile
grads = gradient(() -> loss(data, data), params(m))
```
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.