diff --git a/test.py b/test.py index 33a4f9f3b1e46d3e4cb490ca656f1c9c22e6948e..72ecf7cd071245284e5b58baf88ba4a91bc56f2e 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()