You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I was trying to differentiate the hermiteh polynomial from ClassicalOrthogonalPolynomials.jl using Zygote.gradient and I saw that Zygote hung, started consuming a lot of memory, and then the Julia instance was terminated by the OS. Since there was no error message I am not sure what to investigate. I am attaching my environment information:
pkg> status Zygote ClassicalOrthogonalPolynomials
Status `/tmp/jl_aeo7Zi/Project.toml`
[b30e2e7b] ClassicalOrthogonalPolynomials v0.10.1
[e88e6eb3] Zygote v0.6.62
julia>versioninfo()
Julia Version 1.9.2
Commit e4ee485e909 (2023-07-0509:39 UTC)
Platform Info:
OS: Linux (x86_64-linux-gnu)
CPU:12×Intel(R) Core(TM) i7-8750H CPU @ 2.20GHz
WORD_SIZE:64
LIBM: libopenlibm
LLVM: libLLVM-14.0.6 (ORCJIT, skylake)
Threads:1 on 12 virtual cores
Environment:
LD_PRELOAD =/usr/lib/x86_64-linux-gnu/libstdc++.so.6
JULIA_IMAGE_THREADS =1
Evidently the internals of ClassicalOrthogonalPolynomials are not that AD-friendly. I would recommend either trying different AD libraries or writing AD rules using ChainRulesCore for it. Asking Zygote to handle infinite-length arrays is probably too much!
Hi,
I was trying to differentiate the
hermiteh
polynomial from ClassicalOrthogonalPolynomials.jl usingZygote.gradient
and I saw that Zygote hung, started consuming a lot of memory, and then the Julia instance was terminated by the OS. Since there was no error message I am not sure what to investigate. I am attaching my environment information:This is the MWE
The text was updated successfully, but these errors were encountered: