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