From cfe1f9481597352adf66f4e9254c3601cbf260de Mon Sep 17 00:00:00 2001 From: Matt Wala Date: Tue, 7 Jun 2016 21:36:14 -0500 Subject: [PATCH] Test save/restore of private arrays. --- test/test_loopy.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/test/test_loopy.py b/test/test_loopy.py index 4b6dc898f..5a2243fe6 100644 --- a/test/test_loopy.py +++ b/test/test_loopy.py @@ -1104,13 +1104,16 @@ def test_kernel_splitting_with_loop_and_private_temporary(ctx_factory): knl = lp.make_kernel( "{ [i,k]: 0<=i t_private = a[k,i+1] + <> t_private_scalar = a[k,i+1] + <> t_private_array[i % 2] = a[k,i+1] c[k,i] = a[k,i+1] - out[k,i] = c[k,i] + t_private + out[k,i] = c[k,i] + t_private_scalar + t_private_array[i % 2] """) knl = lp.add_and_infer_dtypes(knl, {"a": np.float32, "c": np.float32, "out": np.float32, "n": np.int32}) + knl = lp.set_temporary_scope(knl, "t_private_scalar", "private") + knl = lp.set_temporary_scope(knl, "t_private_array", "private") ref_knl = knl -- GitLab