From fa490cabf4b0a3723e48fe6bd1519474ee2fbf63 Mon Sep 17 00:00:00 2001
From: Marcel Koch <marcel.koch@uni-muenster.de>
Date: Wed, 30 Jan 2019 14:32:56 +0100
Subject: [PATCH] `identity` removes indices
---
python/dune/codegen/ufl/visitor.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/python/dune/codegen/ufl/visitor.py b/python/dune/codegen/ufl/visitor.py
index e55232d2..ab7c334f 100644
--- a/python/dune/codegen/ufl/visitor.py
+++ b/python/dune/codegen/ufl/visitor.py
@@ -264,6 +264,7 @@ class UFL2LoopyVisitor(ModifiedTerminalTracker):
def identity(self, o):
i, j = self.indices
+ self.indices = None
assert isinstance(i, int) and isinstance(j, int)
return 1 if i == j else 0
--
GitLab