From 4e0fed62453ccb15cf1b40eaeade2970c00d0358 Mon Sep 17 00:00:00 2001
From: Isuru Fernando <isuruf@gmail.com>
Date: Thu, 7 May 2020 23:27:14 -0500
Subject: [PATCH] Fix benchmark
---
benchmarks/bench_translations.py | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/benchmarks/bench_translations.py b/benchmarks/bench_translations.py
index 6e87370f..8a9ef0da 100644
--- a/benchmarks/bench_translations.py
+++ b/benchmarks/bench_translations.py
@@ -66,8 +66,14 @@ class TranslationBenchmarkSuite:
src_rscale = sym.Symbol("src_rscale")
tgt_rscale = sym.Symbol("tgt_rscale")
sac = SymbolicAssignmentCollection()
- result = l_expn.translate_from(m_expn, src_coeff_exprs, src_rscale,
+ try:
+ result = l_expn.translate_from(m_expn, src_coeff_exprs, src_rscale,
dvec, tgt_rscale, sac)
+ except TypeError:
+ # Support older interface to make it possible to compare
+ # in CI run
+ result = l_expn.translate_from(m_expn, src_coeff_exprs, src_rscale,
+ dvec, tgt_rscale)
for i, expr in enumerate(result):
sac.assign_unique("coeff%d" % i, expr)
sac.run_global_cse()
--
GitLab