From acd039a7c56de937e07dd0f0fa24c14ac0c86e0e Mon Sep 17 00:00:00 2001
From: Dominic Kempf <dominic.kempf@iwr.uni-heidelberg.de>
Date: Tue, 8 Nov 2016 17:12:36 +0100
Subject: [PATCH] Add handler for math functions
---
python/dune/perftool/loopy/mangler.py | 14 ++++++++++++++
python/dune/perftool/pdelab/localoperator.py | 3 +++
2 files changed, 17 insertions(+)
create mode 100644 python/dune/perftool/loopy/mangler.py
diff --git a/python/dune/perftool/loopy/mangler.py b/python/dune/perftool/loopy/mangler.py
new file mode 100644
index 00000000..857052e6
--- /dev/null
+++ b/python/dune/perftool/loopy/mangler.py
@@ -0,0 +1,14 @@
+""" Function manglers for math functions in C++ """
+
+from dune.perftool.generation import function_mangler
+
+from loopy import CallMangleInfo
+
+
+@function_mangler
+def dune_math_manglers(kernel, name, arg_dtypes):
+ if name == "exp":
+ return CallMangleInfo("std::exp",
+ arg_dtypes,
+ arg_dtypes,
+ )
diff --git a/python/dune/perftool/pdelab/localoperator.py b/python/dune/perftool/pdelab/localoperator.py
index e214b523..4359f7dd 100644
--- a/python/dune/perftool/pdelab/localoperator.py
+++ b/python/dune/perftool/pdelab/localoperator.py
@@ -26,6 +26,9 @@ from dune.perftool.cgen.clazz import (AccessModifier,
ClassMember,
)
from dune.perftool.ufl.modified_terminals import Restriction
+
+import dune.perftool.loopy.mangler
+
from pymbolic.primitives import Variable
from pytools import Record
--
GitLab