Closed pzimbrod closed 2 years ago
Maybe related to a reported problem that Zygote has with differentiating FFTs.
Possible fix could be to work with @view
and do in-place transforms as suggested in the linked Discourse thread.
Found a workaround for now. Closing.
When running
burgers.jl
with commit c62d8deb2efb72b11fb9b903720f6615e4ea93b6, Julia allocates massive amounts of memory to a point that Julia gets killed on a 128GB RAM machine.This behavior is abnormal and likely somewhere an abundant amount of copies takes place. This is reproducible when running both on CPU and GPU.