Merge pull request #571 from OP2/wence/fix/configure-flop-log

kernel: Add configuration parameter to control kernel.num_flops
OP2 · Oct 23, 2019 · a213238 · a213238
2 parents 070367e + ba54157
commit a213238
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/pyop2/base.py b/pyop2/base.py
@@ -3371,6 +3371,8 @@ def code(self):
 
     @cached_property
     def num_flops(self):
+        if not configuration["compute_kernel_flops"]:
+            return 0
         if isinstance(self.code, Node):
             v = EstimateFlops()
             return v.visit(self.code)

diff --git a/pyop2/configuration.py b/pyop2/configuration.py
@@ -79,6 +79,7 @@ class Configuration(dict):
         "debug": ("PYOP2_DEBUG", bool, False),
         "cflags": ("PYOP2_CFLAGS", str, ""),
         "ldflags": ("PYOP2_LDFLAGS", str, ""),
+        "compute_kernel_flops": ("PYOP2_COMPUTE_KERNEL_FLOPS", bool, False),
         "type_check": ("PYOP2_TYPE_CHECK", bool, True),
         "check_src_hashes": ("PYOP2_CHECK_SRC_HASHES", bool, True),
         "log_level": ("PYOP2_LOG_LEVEL", (str, int), "WARNING"),