From 0a55895d4f04f28dabf5cd684a9ec9163f36a6ef Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 27 May 2016 10:22:30 -0500
Subject: [PATCH] Some test simplifications

---
 test/test_linalg.py |  2 +-
 test/test_loopy.py  | 16 +++++++---------
 2 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/test/test_linalg.py b/test/test_linalg.py
index 0e0b59089..e3cae8557 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 613f6547e..f0718fbab 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()
 
-- 
GitLab