unrelated, start adding an option to turn CSE off

mcabbott · Dec 18, 2020 · 1d1221d · 1d1221d
1 parent e237ee6
commit 1d1221d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/symbolic.jl b/src/symbolic.jl
@@ -3,6 +3,8 @@
 
 using DiffRules
 
+const _CSE = Ref(true)
+
 function insert_symbolic_gradient(axislist, store)
 
     dZ = Symbol(DEL, ZED)
@@ -50,7 +52,7 @@ function insert_symbolic_gradient(axislist, store)
         end
     end
     store.verbose>0 && @info "symbolic gradients" inbody
-    ex_body = commonsubex(quote $(inbody...) end)
+    ex_body = _CSE[] ? commonsubex(quote $(inbody...) end) : quote $(inbody...) end
 
     ex_pre, ex_post = if store.redfun == :* # then nonzero LHS are handled already, but harder cases here:
         product_grad(prebody, store)