diff --git a/loopy/statistics.py b/loopy/statistics.py index 240bef8eb4d361858b6791f11d4e25b804777a74..1d882ce88df38f84e72647d7018ceb6cd0b21ed5 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 dbfd683f736bd773f2fc280ee64b41976b1efb80..a504761193fe4acb7dff9a4a9535efb7a74fe2a9 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():