diff --git a/python/dune/perftool/sumfact/accumulation.py b/python/dune/perftool/sumfact/accumulation.py index 2baac3955129376b269fb5a9b9c10b17dcf3a8fa..50f4d169935cae6ec421da62ba432506d055f298 100644 --- a/python/dune/perftool/sumfact/accumulation.py +++ b/python/dune/perftool/sumfact/accumulation.py @@ -201,6 +201,7 @@ def generate_accumulation_instruction(expr, visitor): # Number of basis functions per direction leaf_element = test_info.element + from ufl import MixedElement if isinstance(leaf_element, MixedElement): leaf_element = leaf_element.extract_component(test_info.element_index)[1] degree = leaf_element._degree diff --git a/python/dune/perftool/sumfact/basis.py b/python/dune/perftool/sumfact/basis.py index 4e1237c5d6c4e7b5d012de1d9328b06e6d350952..8e745c7daf65294db5e9f34b03c7a19736918f8c 100644 --- a/python/dune/perftool/sumfact/basis.py +++ b/python/dune/perftool/sumfact/basis.py @@ -42,7 +42,7 @@ from dune.perftool.tools import maybe_wrap_subscript from dune.perftool.pdelab.basis import shape_as_pymbolic from dune.perftool.sumfact.accumulation import sumfact_iname -from ufl import VectorElement, TensorElement, TensorProductElement +from ufl import MixedElement, VectorElement, TensorElement, TensorProductElement from pytools import product, ImmutableRecord diff --git a/python/dune/perftool/sumfact/tabulation.py b/python/dune/perftool/sumfact/tabulation.py index 0ae2b8327f6112d15c80b439680da03d25e70d6c..e8518668c227c7528ab72712bbd9e2cba92c147f 100644 --- a/python/dune/perftool/sumfact/tabulation.py +++ b/python/dune/perftool/sumfact/tabulation.py @@ -208,7 +208,7 @@ def quadrature_points_per_direction(): q = (q,) * world_dimension() # Quadrature points in per direction - nb_qp = tuple(order // 2 +1 for order in q) + nb_qp = tuple(order // 2 + 1 for order in q) return nb_qp diff --git a/python/dune/perftool/ufl/visitor.py b/python/dune/perftool/ufl/visitor.py index 70b95dbc30bae42848ebd8e7e30c494ef7a8c59b..31e39fe8ff9ebbbce1ce0bf44fc14448ec44ee71 100644 --- a/python/dune/perftool/ufl/visitor.py +++ b/python/dune/perftool/ufl/visitor.py @@ -23,6 +23,7 @@ from pymbolic.primitives import (Call, from ufl.algorithms import MultiFunction from ufl.checks import is_cellwise_constant from ufl import (VectorElement, + MixedElement, TensorElement, TensorProductElement, )