From 0de948d37beca09eaacfe6506d64a5bcce4d3826 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner <inform@tiker.net> Date: Fri, 24 Aug 2018 23:26:33 -0400 Subject: [PATCH] Fix work remover test --- test/test_transform.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test/test_transform.py b/test/test_transform.py index ed00ebd1b..ae0a577c1 100644 --- a/test/test_transform.py +++ b/test/test_transform.py @@ -553,6 +553,8 @@ def test_remove_work(ctx_factory): ], assumptions="n>=1") + knl = lp.add_and_infer_dtypes(knl, dict(a=np.float32)) + knl = lp.split_iname(knl, "i", 16, outer_tag="g.1", inner_tag="l.1") knl = lp.split_iname(knl, "j", 16, outer_tag="g.0", inner_tag="l.0") knl = lp.add_prefetch(knl, "a", ["i_inner", "j_inner"], @@ -562,7 +564,8 @@ def test_remove_work(ctx_factory): from loopy.transform.instruction import remove_work knl = remove_work(knl) - lp.auto_test_vs_ref(None, ctx, knl, print_ref_code=False) + lp.auto_test_vs_ref(knl, ctx, None, print_ref_code=False, + parameters=dict(n=512)) if __name__ == "__main__": -- GitLab