From 4aab42a166cf11745b50250de37c41714aa6c3f8 Mon Sep 17 00:00:00 2001
From: Dominic Kempf <dominic.kempf@iwr.uni-heidelberg.de>
Date: Tue, 25 Oct 2016 17:43:26 +0200
Subject: [PATCH] Start introducing backend calls
---
python/dune/perftool/pdelab/argument.py | 4 ++--
python/dune/perftool/pdelab/basis.py | 4 +++-
python/dune/perftool/pdelab/localoperator.py | 8 +++++---
python/dune/perftool/pdelab/quadrature.py | 5 +++++
python/dune/perftool/sumfact/sumfact.py | 6 ++++--
5 files changed, 19 insertions(+), 8 deletions(-)
diff --git a/python/dune/perftool/pdelab/argument.py b/python/dune/perftool/pdelab/argument.py
index 0e81542c..76639a23 100644
--- a/python/dune/perftool/pdelab/argument.py
+++ b/python/dune/perftool/pdelab/argument.py
@@ -103,7 +103,7 @@ def name_trialfunction(element, restriction, component):
rawname = "u" + "_".join(str(c) for c in component)
name = restricted_name(rawname, restriction)
container = name_coefficientcontainer(restriction)
- evaluate_coefficient(element, name, container, restriction, component)
+ get_backend("eval_coefficient")(element, name, container, restriction, component)
return name
@@ -119,7 +119,7 @@ def name_apply_function(element, restriction, component):
rawname = "z_func" + "_".join(str(c) for c in component)
name = restricted_name(rawname, restriction)
container = name_applycontainer(restriction)
- evaluate_coefficient(element, name, container, restriction, component)
+ get_backend("eval_coefficient")(element, name, container, restriction, component)
return name
diff --git a/python/dune/perftool/pdelab/basis.py b/python/dune/perftool/pdelab/basis.py
index 09ecddb0..ad587091 100644
--- a/python/dune/perftool/pdelab/basis.py
+++ b/python/dune/perftool/pdelab/basis.py
@@ -1,6 +1,7 @@
""" Generators for basis evaluations """
-from dune.perftool.generation import (cached,
+from dune.perftool.generation import (backend,
+ cached,
class_member,
generator_factory,
include_file,
@@ -123,6 +124,7 @@ def shape_as_pymbolic(shape):
return tuple(_shape_as_pymbolic(s) for s in shape)
+@backend(interface="eval_coefficient")
@cached
def evaluate_coefficient(element, name, container, restriction, component):
from ufl.functionview import select_subelement
diff --git a/python/dune/perftool/pdelab/localoperator.py b/python/dune/perftool/pdelab/localoperator.py
index 7a299ab7..f456b8d8 100644
--- a/python/dune/perftool/pdelab/localoperator.py
+++ b/python/dune/perftool/pdelab/localoperator.py
@@ -403,11 +403,13 @@ def generate_accumulation_instruction(visitor, accterm, measure, subdomain_id):
(ansatz_lfs.get_args() + test_lfs.get_args() + (pymbolic_expr,))
)
- from dune.perftool.generation import instruction
- from dune.perftool.pdelab.quadrature import quadrature_iname
+ from dune.perftool.generation import get_backend, instruction
+ from dune.perftool.options import option_switch
+ quad_inames = get_backend(interface="quadinames", selector=option_switch("sumfac"))()
+
instruction(assignees=(),
expression=expr,
- forced_iname_deps=additional_inames.union(frozenset(visitor.inames).union(frozenset({quadrature_iname()}))),
+ forced_iname_deps=additional_inames.union(frozenset(visitor.inames).union(quad_inames)),
forced_iname_deps_is_final=True,
predicates=predicates
)
diff --git a/python/dune/perftool/pdelab/quadrature.py b/python/dune/perftool/pdelab/quadrature.py
index b778de54..b50c8248 100644
--- a/python/dune/perftool/pdelab/quadrature.py
+++ b/python/dune/perftool/pdelab/quadrature.py
@@ -17,6 +17,11 @@ def quadrature_iname():
return "q"
+@backend(interface="quadinames")
+def quadrature_inames():
+ return frozenset({quadrature_iname()})
+
+
def quadrature_preamble(code, **kw):
return instruction(inames=quadrature_iname(), code=code, **kw)
diff --git a/python/dune/perftool/sumfact/sumfact.py b/python/dune/perftool/sumfact/sumfact.py
index a7ee0a04..5a3e364d 100644
--- a/python/dune/perftool/sumfact/sumfact.py
+++ b/python/dune/perftool/sumfact/sumfact.py
@@ -1,5 +1,6 @@
from dune.perftool.pdelab.argument import pymbolic_coefficient
-from dune.perftool.generation import (domain,
+from dune.perftool.generation import (backend,
+ domain,
get_counter,
iname,
instruction,
@@ -60,7 +61,8 @@ def setup_theta(element, container, restriction, component, a_matrices):
# TODO this code is WIP and mainly used for experiments.
-def start_sumfactorization(element, container, restriction, component):
+@backend(interface="eval_coefficient", name="sumfact")
+def start_sumfactorization(element, name, container, restriction, component):
theta = name_theta()
rows = quadrature_points_per_direction()
cols = basis_functions_per_direction()
--
GitLab