Closed CarloLucibello closed 1 year ago
The regression happened with Zygote v0.6.44
Looks like the problem is due to boxing indeed and can be considered a duplicate of #290. I'll fix things on the GNN.jl side avoiding closures. Thanks for looking into it @ToucheSir
A recent update to Zygote or ChainRules has caused a test failure in GNN.jl. I had a hard time trying to reduce the example, e.g. same code but within a function triggers the error, so I'll just post the opaque example below hoping that someone can decrypt the error. This is on
The gradient of the NNConv layer now fails. The layer involves
NNlib.batched_mul
, which seems to be a crucial ingredient for triggering the error.Reproducer: