implement evalf_iadd for Inflate

This patch implements `evalf_iadd` for `Inflate`.
evalf · Mar 14, 2024 · c9623e6 · c9623e6
1 parent 6d30a9b
commit c9623e6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/nutils/evaluable.py b/nutils/evaluable.py
@@ -3025,6 +3025,9 @@ def evalf(self, array, indices, length):
         numpy.add.at(inflated, self._indices_head + (indices,), array)
         return inflated
 
+    def evalf_iadd(self, output, array, indices, length):
+        numpy.add.at(output, self._indices_head + (indices,), array)
+
     def _inflate(self, dofmap, length, axis):
         if dofmap.ndim == 0 and dofmap == self.dofmap and length == self.length:
             return diagonalize(self, -1, axis)