Set more functions as non-differentiable

bsc-quantic · Mar 28, 2024 · dcf9808 · dcf9808
1 parent da24bf8
commit dcf9808
Showing 1 changed file with 18 additions and 1 deletion.
diff --git 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