From 7ee2a3e3f186142cca88d27ca44c45e8eadc5f9c Mon Sep 17 00:00:00 2001
From: Dominic Kempf <dominic.kempf@iwr.uni-heidelberg.de>
Date: Wed, 30 Aug 2017 15:23:33 +0200
Subject: [PATCH] Take into account non-square blocks in the jacobian!
---
python/dune/perftool/sumfact/realization.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/python/dune/perftool/sumfact/realization.py b/python/dune/perftool/sumfact/realization.py
index 96ef0012..abefc185 100644
--- a/python/dune/perftool/sumfact/realization.py
+++ b/python/dune/perftool/sumfact/realization.py
@@ -228,12 +228,16 @@ def _realize_sum_factorization_kernel(sf):
direct_output = "{}x{}".format(direct_output, sf.trial_element_index)
rowsize = sum(tuple(s for s in _local_sizes(sf.trial_element)))
+ element = sf.trial_element
+ if element.num_sub_elements() > 0:
+ element = element.extract_component(sf.trial_element_index)[1]
+ other_shape = tuple(element.degree() + 1 for e in range(sf.length))
from pytools import product
manual_strides = tuple("stride:{}".format(rowsize * product(output_shape[:i])) for i in range(sf.length))
dim_tags = "{},{}".format(novec_ftags, ",".join(manual_strides))
globalarg(direct_output,
dtype=np.float64,
- shape=output_shape + output_shape,
+ shape=other_shape + output_shape,
offset=rowsize * _dof_offset(sf.test_element, sf.test_element_index) + _dof_offset(sf.trial_element, sf.trial_element_index),
dim_tags=dim_tags,
)
--
GitLab