diff --git a/ext/EinExprsChainRulesCoreExt.jl b/ext/EinExprsChainRulesCoreExt.jl index b5ae37f..e8b7882 100644 --- a/ext/EinExprsChainRulesCoreExt.jl +++ b/ext/EinExprsChainRulesCoreExt.jl @@ -3,6 +3,23 @@ module EinExprsChainRulesCoreExt using EinExprs using ChainRulesCore -@non_differentiable einexpr(::Any...) +for f in [ + :head, + :args, + :nargs, + :inds, + :branches, + :leaves, + :suminds, + :parsuminds, + :einexpr, + :sumtraces, + :indshistogram, + :hyperinds, + :neighbours, + :select, +] + @eval @non_differentiable EinExprs.$f(::Any...) +end end \ No newline at end of file