Select components of list tensors if indexed with fixed indices

python/dune/perftool/ufl/visitor.py

@@ -232,7 +232,12 @@ class UFL2LoopyVisitor(ModifiedTerminalTracker):
         return self._index_or_fixed_index(o)
 
     def list_tensor(self, o):
-        return self.interface.pymbolic_list_tensor(o)
+        if all(isinstance(i, int) for i in self.indices):
+            index = self.indices[0]
+            self.indices = self.indices[1:]
+            return self.call(o.ufl_operands[index])
+        else:
+            return self.interface.pymbolic_list_tensor(o)
 
     def identity(self, o):
         return self.interface.pymbolic_identity(o)