diff --git a/test/test_linalg.py b/test/test_linalg.py index 0e0b59089fe1d3a1c1310bd0834a29ca751b8df0..e3cae8557a2266bb7b27adb4d6148ed2cda32063 100644 --- a/test/test_linalg.py +++ b/test/test_linalg.py @@ -604,7 +604,7 @@ def test_small_batched_matvec(ctx_factory): Np = 36 # noqa knl = lp.make_kernel( - "[K] -> {[i,j,k]: 0<=k<K and 0<= i,j < %d}" % Np, + "{[i,j,k]: 0<=k<K and 0<= i,j < %d}" % Np, [ "result[k, i] = sum(j, d[i, j]*f[k, j])" ], diff --git a/test/test_loopy.py b/test/test_loopy.py index 613f6547ebf96bc48f5b6da49af53483b4563621..f0718fbab68450227c377afcaf383cab5db5cacb 100644 --- a/test/test_loopy.py +++ b/test/test_loopy.py @@ -572,7 +572,6 @@ def test_fuzz_code_generator(ctx_factory): def test_empty_reduction(ctx_factory): - dtype = np.dtype(np.float32) ctx = ctx_factory() queue = cl.CommandQueue(ctx) @@ -581,15 +580,14 @@ def test_empty_reduction(ctx_factory): "{[i]: 0<=i<20}", "[i] -> {[j]: 0<=j<0}" ], - [ - "a[i] = sum(j, j)", - ], - [ - lp.GlobalArg("a", dtype, (20,)), - ]) - cknl = lp.CompiledKernel(ctx, knl) + "a[i] = sum(j, j)", + ) + + knl = lp.realize_reduction(knl) + print(knl) - evt, (a,) = cknl(queue) + knl = lp.set_options(knl, write_cl=True) + evt, (a,) = knl(queue) assert (a.get() == 0).all()