From 931814858c6bc2b9866af9370e587addac568825 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 24 May 2019 17:55:05 -0500
Subject: [PATCH] Simplify matvec check
---
test.py | 5 +----
1 file changed, 1 insertion(+), 4 deletions(-)
diff --git a/test.py b/test.py
index 33a4f9f..72ecf7c 100644
--- a/test.py
+++ b/test.py
@@ -46,17 +46,14 @@ def test_matvec(ctx_factory):
prg = get_weno_program()
prg = with_root_kernel(prg, "mult_mat_vec")
- print(prg)
a = np.random.rand(10, 10).astype(np.float32).copy(order="F")
b = np.random.rand(10).astype(np.float32)
- a_dev = cl.array.to_device(queue, a)
- b_dev = cl.array.to_device(queue, b)
c_dev = cl.array.empty(queue, 10, dtype=np.float32)
prg = lp.set_options(prg, write_cl=True)
- prg(queue, a=a_dev, b=b_dev, c=c_dev, alpha=1.0)
+ prg(queue, a=a, b=b, c=c_dev, alpha=1.0)
c = c_dev.get()
--
GitLab