diff --git a/examples/matrix-ops.py b/examples/matrix-ops.py index 30f01f6c5a448cc3f33a157a3207037ea2d889ff..abfa7c0c1407063f10b88af755c39324252b3da2 100644 --- a/examples/matrix-ops.py +++ b/examples/matrix-ops.py @@ -26,19 +26,16 @@ def plain_matrix_mul(ctx_factory=cl.create_some_context): a, b, c, i, j, k, n_sym = [var(s) for s in "abcijkn"] knl = lp.LoopKernel(ctx.devices[0], - [ - lp.LoopDimension("i", n), - lp.LoopDimension("j", n), - lp.LoopDimension("k", n), - ], [ - (c[i, j], a[i, k]*b[k, j]) - ], - [ - lp.ArrayArg("a", dtype, shape=(n, n)), - lp.ArrayArg("b", dtype, shape=(n, n)), - lp.ArrayArg("c", dtype, shape=(n, n)), - ], - name="matmul") + "[n] -> {[i,j,k]: 0<=i,j,k