This might be considered an edge case because of the Boolean type, but I'm leaving it here as an FYI in case it turns out to matter.
This used to work (gives gradient of nothing) under 1.9.0, but under 1.10 it does not.
I suspect this might apply more broadly to anything declared with @non_differentiable and might point to a change needed in that macro in ChainRulesCore.
This might be considered an edge case because of the Boolean type, but I'm leaving it here as an FYI in case it turns out to matter. This used to work (gives gradient of
nothing
) under 1.9.0, but under 1.10 it does not.MWE:
System info
side note - this 1.10 is my own build with 2 lines differing from 1.10 in linear algebra, nothing to do with Flux, sum, mean....