composed map: add permute to kwarg

OP2 · Jun 14, 2024 · acae6c5 · acae6c5
1 parent a52b998
commit acae6c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyop2/codegen/builder.py b/pyop2/codegen/builder.py
@@ -190,13 +190,13 @@ def __init__(self, *maps_):
         self.offset = maps_[0].offset
         self.maps_ = maps_
 
-    def indexed(self, multiindex, layer=None):
+    def indexed(self, multiindex, layer=None, permute=lambda x: x):
         n, i, f = multiindex
         n_ = n
         for map_ in reversed(self.maps_):
             if map_ is not self.maps_[0]:
                 n_, (_, _) = map_.indexed(MultiIndex(n_, FixedIndex(0), Index()), layer=None)
-        return self.maps_[0].indexed(MultiIndex(n_, i, f), layer=layer)
+        return self.maps_[0].indexed(MultiIndex(n_, i, f), layer=layer, permute=permute)
 
 
 class Pack(metaclass=ABCMeta):