From a2501f35fb9770cc4b0716503594a47d46f389b6 Mon Sep 17 00:00:00 2001
From: James Stevens <jdsteve2@illinois.edu>
Date: Tue, 23 Jun 2015 18:20:56 -0500
Subject: [PATCH] modifying bitwise op count test after fixing bitwise op
counters
---
test/test_statistics.py | 22 ++++++++++------------
1 file changed, 10 insertions(+), 12 deletions(-)
diff --git a/test/test_statistics.py b/test/test_statistics.py
index b9ec0a1c4..5f276617d 100644
--- a/test/test_statistics.py
+++ b/test/test_statistics.py
@@ -140,21 +140,19 @@ def test_op_counter_bitwise():
knl = lp.add_and_infer_dtypes(
knl, dict(
a=np.int32, b=np.int32,
- g=np.int64, h=np.int64))
+ g=np.int32, h=np.int32))
poly = get_op_poly(knl)
-
- n = 10
- m = 10
- l = 10
- param_values = {'n': n, 'm': m, 'l': l}
- i32 = poly.dict[np.dtype(np.int32)].eval_with_dict(param_values)
- i64 = poly.dict[np.dtype(np.int64)].eval_with_dict(param_values)
- not_there = poly[np.dtype(np.float64)].eval_with_dict(param_values)
+ n = 512
+ m = 256
+ l = 128
+ i32 = poly.dict[np.dtype(np.int32)].eval_with_dict({'n': n, 'm': m, 'l': l})
+ # i64 = poly.dict[np.dtype(np.int64)].eval_with_dict({'n': n, 'm': m, 'l': l}) # noqa
print(poly.dict)
- assert i32 == n*m + n*m*l
- assert i64 == 2*n*m
- assert not_there == 0
+ f64 = poly[np.dtype(np.float64)].eval_with_dict({'n': n, 'm': m, 'l': l})
+ assert i32 == 7*n*m+3*n*m*l
+ assert f64 == 0
+ # TODO test bitwise operations on int64
def test_op_counter_triangular_domain():
--
GitLab