From 605244e3f3fc9965eabdec8850ff3a5f43e42f34 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner <inform@tiker.net> Date: Mon, 24 Aug 2015 22:08:13 -0500 Subject: [PATCH] More follow-up fixes for 6fb2849ef --- loopy/statistics.py | 2 +- test/test_statistics.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/loopy/statistics.py b/loopy/statistics.py index 240bef8eb..1d882ce88 100755 --- a/loopy/statistics.py +++ b/loopy/statistics.py @@ -595,4 +595,4 @@ def get_barrier_poly(knl): else: barrier_poly += isl.PwQPolynomial('{ 1 }') - return barrier_poly.dict + return barrier_poly diff --git a/test/test_statistics.py b/test/test_statistics.py index dbfd683f7..a50476119 100644 --- a/test/test_statistics.py +++ b/test/test_statistics.py @@ -148,10 +148,9 @@ def test_op_counter_bitwise(): l = 128 i32 = poly[np.dtype(np.int32)].eval_with_dict({'n': n, 'm': m, 'l': l}) i64 = poly[np.dtype(np.int64)].eval_with_dict({'n': n, 'm': m, 'l': l}) # noqa - f64 = poly[np.dtype(np.float64)].eval_with_dict({'n': n, 'm': m, 'l': l}) + assert np.dtype(np.float64) not in poly assert i32 == n*m+3*n*m*l assert i64 == 6*n*m - assert f64 == 0 def test_op_counter_triangular_domain(): -- GitLab