k

Author: ksagiyam

ufl/algorithms/apply_derivatives.py

@@ -2021,10 +2021,11 @@ def compute_gprimeterm(ngrads, vval, vcomp, wshape, wcomp):
         return gprimesum
 
     @process.register(CoordinateDerivative)
-    def _(self, o: Expr) -> Expr:
+    @DAGTraverser.postorder_only_children([0])
+    def _(self, o: Expr, o0) -> Expr:
         """Differentiate a coordinate_derivative."""
-        o = o.ufl_operands
-        return CoordinateDerivative(map_expr_dag(self, o[0]), o[1], o[2], o[3])
+        _, o1, o2, o3 = o.ufl_operands
+        return CoordinateDerivative(o0, o1, o2, o3)
 
     @process.register(BaseFormOperator)
     @DAGTraverser.postorder

ufl/corealg/dag_traverser.py

@@ -95,3 +95,23 @@ def wrapper(self, o, *args):
             return method(self, o, *processed_operands, *args)
 
         return wrapper
+
+    @staticmethod
+    def postorder_only_children(indices):
+        """Insert processed operands after ``o`` and before ``*args``.
+
+        Users can write a post-order singledispatch method that takes
+        ``o``, processed operands, and additional arguments ``*args`` (if any),
+        wrappong them in this decorator.
+
+        """
+
+        def postorder(method):
+            @wraps(method)
+            def wrapper(self, o, *args):
+                processed_operands = [self(o.ufl_operands[i]) for i in indices]
+                return method(self, o, *processed_operands, *args)
+
+            return wrapper
+
+        return postorder