When looking at SnoopCompile timings for Zygote (e.g.), IRTools comes up as a big inference block. AIUI this is because Zygote itself can't precompile any methods in IRTools because it does not own them, and hence nothing precompiles reliably all the way up the chain.
When looking at SnoopCompile timings for Zygote (e.g.), IRTools comes up as a big inference block. AIUI this is because Zygote itself can't precompile any methods in IRTools because it does not own them, and hence nothing precompiles reliably all the way up the chain.
https://github.com/FluxML/IRTools.jl/issues/64 may be related.