From 92f45555f85a21166c5da698fd6a17682256cefe Mon Sep 17 00:00:00 2001
From: Dominic Kempf <dominic.kempf@iwr.uni-heidelberg.de>
Date: Thu, 28 Jul 2016 15:40:24 +0200
Subject: [PATCH] [bugfix] recognize the shape of coefficients in dimension
index aliasing
---
python/dune/perftool/ufl/dimensionindex.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/python/dune/perftool/ufl/dimensionindex.py b/python/dune/perftool/ufl/dimensionindex.py
index e7c48a25..e5fb3e9f 100644
--- a/python/dune/perftool/ufl/dimensionindex.py
+++ b/python/dune/perftool/ufl/dimensionindex.py
@@ -22,6 +22,9 @@ class _DimensionIndexMapping(MultiFunction):
subelement = select_subelement(o.ufl_operands[0].ufl_element(), o.ufl_operands[1])
self.shape = len(subelement.value_shape())
+ def coefficient(self, o):
+ self.shape = len(o.ufl_element().value_shape())
+
def indexed(self, o):
self.call(o.ufl_operands[0])
for i in range(self.shape):
--
GitLab