From 4f9b02ae8a0f743a220063ea24f5e00104a7b2f8 Mon Sep 17 00:00:00 2001
From: Matt Wala <wala1@illinois.edu>
Date: Tue, 10 Jan 2017 09:13:52 -0600
Subject: [PATCH] Add a timing message.
---
sumpy/assignment_collection.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/sumpy/assignment_collection.py b/sumpy/assignment_collection.py
index 2bfdbc2a..840b04da 100644
--- a/sumpy/assignment_collection.py
+++ b/sumpy/assignment_collection.py
@@ -159,6 +159,9 @@ class SymbolicAssignmentCollection(object):
return new_name
def run_global_cse(self, extra_exprs=[]):
+ import time
+ start_time = time.time()
+
logger.info("common subexpression elimination: start")
assign_names = sorted(self.assignments)
@@ -185,7 +188,8 @@ class SymbolicAssignmentCollection(object):
assert isinstance(name, sp.Symbol)
self.add_assignment(name.name, value)
- logger.info("common subexpression elimination: done")
+ logger.info("common subexpression elimination: done after {dur:.2f} s"
+ .format(dur=time.time() - start_time))
return new_extra_exprs
def kill_trivial_assignments(self, exprs):
--
GitLab