In-place assignment

Author: jrmaddison

firedrake/adjoint_utils/constant.py

@@ -104,6 +104,9 @@ def _ad_assign_numpy(dst, src, offset):
     def _ad_to_list(m):
         return m.dat.data_ro.reshape(-1).tolist()
 
+    def _ad_assign(self, other):
+        self.assign(other)
+
     def _ad_copy(self):
         return self._constant_from_values()
 

firedrake/adjoint_utils/function.py

@@ -352,6 +352,9 @@ def _ad_to_list(m):
 
         return m_a.tolist()
 
+    def _ad_assign(self, other):
+        self.assign(other)
+
     def _ad_copy(self):
         from firedrake import Function