diff --git a/test/test_linalg.py b/test/test_linalg.py index 6cad0b0804849ac11fe3e36e3771a91bb104e225..de14bac1e6fd22dceeadfbd91a7d376400c4b206 100644 --- a/test/test_linalg.py +++ b/test/test_linalg.py @@ -191,14 +191,14 @@ def test_plain_matrix_mul(ctx_factory): n = get_suitable_size(ctx) - for dtype, check, vec_size, reduction_func in [ - (cl_array.vec.float4, check_float4, 4, "sum_vec_float4"), - (np.float32, None, 1, "sum_float32"), + for dtype, check, vec_size in [ + (cl_array.vec.float4, check_float4, 4), + (np.float32, None, 1), ]: knl = lp.make_kernel(ctx.devices[0], "{[i,j,k]: 0<=i,j,k<%d}" % n, [ - "c[i, j] = %s(k, a[i, k]*b[k, j])" % reduction_func + "c[i, j] = sum(k, a[i, k]*b[k, j])" ], [ lp.GlobalArg("a", dtype, shape=(n, n), order=order), @@ -232,15 +232,13 @@ def test_variable_size_matrix_mul(ctx_factory): dtype = np.float32 ctx = ctx_factory() order = "C" - queue = cl.CommandQueue(ctx, - properties=cl.command_queue_properties.PROFILING_ENABLE) n = get_suitable_size(ctx) knl = lp.make_kernel(ctx.devices[0], "[n] -> {[i,j,k]: 0<=i,j,k {[i,j,k]: 0<=i,j,k