diff --git a/examples/sem_reagan.py b/examples/sem_reagan.py index 88dde57a6c5927ecb6f13bd0a6ce213008438fc9..42c3dcfec32907dd66cb125a0a51776636fcc720 100644 --- a/examples/sem_reagan.py +++ b/examples/sem_reagan.py @@ -35,129 +35,6 @@ from pyopencl.tools import pytest_generate_tests_for_pyopencl \ -def test_tim2d(ctx_factory): - dtype = np.float32 - ctx = ctx_factory() - order = "C" - - n = 8 - - from pymbolic import var - K_sym = var("K") - - field_shape = (K_sym, n, n) - - # K - run-time symbolic - knl = lp.make_kernel(ctx.devices[0], - "[K] -> {[i,j,e,m,o,gi]: 0<=i,j,m,o<%d and 0<=e {[i,j,k, o, e]: 0<=i,j,k,o < n and 0<=e ur[i,j,k] = sum(@o, D[i,o]*u[e,o,j,k]) - <> us[i,j,k] = sum(@o, D[j,o]*u[e,i,o,k]) - <> ut[i,j,k] = sum(@o, D[k,o]*u[e,i,j,o]) - """, - [ - lp.GlobalArg("u", dtype, shape="E,n,n,n", order=order), - # lp.GlobalArg("G", dtype, shape=(3,)+field_shape, order=order), - # lp.ConstantArrayArg("D", dtype, shape=(n, n), order=order), - lp.GlobalArg("D", dtype, shape=(n, n), order=order), - # lp.ImageArg("D", dtype, shape=(n, n)), - lp.ValueArg("E", np.int32, approximately=1000), - ], - name="semdiff3D", assumptions="E>=1", - defines={"n": n}) - - seq_knl = knl - - def variant_orig(knl): - return knl - - for variant in [variant_orig]: - kernel_gen = lp.generate_loop_schedules(variant(knl)) - kernel_gen = lp.check_kernels(kernel_gen, dict(K=1000)) - - E = 1000 - lp.auto_test_vs_ref(seq_knl, ctx, kernel_gen, - op_count=[E*(n*n*n*2*2 + n*n*2*3 + n**3 * 2*2)/1e9], - op_label=["GFlops"], - parameters={"E": E}) - - - - def test_tim3d_slab(ctx_factory): dtype = np.float32 ctx = ctx_factory() @@ -165,9 +42,6 @@ def test_tim3d_slab(ctx_factory): Nq = 8 - from pymbolic import var - - # K - run-time symbolic knl = lp.make_kernel(ctx.devices[0], "[E] -> {[i,j, k, o,m, e]: 0<=i,j,k,o,m < Nq and 0<=e