PastaQ.jl icon indicating copy to clipboard operation
PastaQ.jl copied to clipboard

Zygote `rrule(typeof(apply),...)` performance drop

Open GTorlai opened this issue 3 years ago • 2 comments

Looks like the time-to-solution to obtain gradients from TEBD simulation of dynamics (i.e. OCC) have substantial overhead compare to simple evaluation of the function.

GTorlai avatar Apr 07 '22 20:04 GTorlai

Could you share a benchmark code and some timings? What's the ratio of the time to compute the gradient to the time of just computing the function?

mtfishman avatar Apr 07 '22 20:04 mtfishman

This performance drop has been reported by a user. I will work to produce a minimal example. I just opened this not to forget.

GTorlai avatar Apr 07 '22 21:04 GTorlai