diff --git a/benchmark.py b/benchmark.py index 71859c74a50af454451a47b6bfa2a176114c66fb..6d2b9b007f6af52bcfd55906658879ee5764219c 100644 --- a/benchmark.py +++ b/benchmark.py @@ -9,16 +9,16 @@ import sys import logging -from utilities import * +import utilities as u def benchmark_compute_flux_derivatives_gpu(ctx_factory, write_code=False): logging.basicConfig(level="INFO") - prg = get_weno_program() - prg = transform_weno_for_gpu(prg) + prg = u.get_weno_program() + prg = u.transform_weno_for_gpu(prg) - queue = get_queue(ctx_factory) + queue = u.get_queue(ctx_factory) prg = prg.copy(target=lp.PyOpenCLTarget(queue.device)) prg = lp.set_options(prg, no_numpy=True) prg = lp.set_options(prg, ignore_boostable_into=True) @@ -34,16 +34,16 @@ def benchmark_compute_flux_derivatives_gpu(ctx_factory, write_code=False): nz = n print("ARRAY GEN") - states = random_array_on_device(queue, nvars, nx+6, ny+6, nz+6) - fluxes = random_array_on_device(queue, nvars, ndim, nx+6, ny+6, nz+6) - metrics = random_array_on_device(queue, ndim, ndim, nx+6, ny+6, nz+6) - metric_jacobians = random_array_on_device(queue, nx+6, ny+6, nz+6) + states = u.random_array_on_device(queue, nvars, nx+6, ny+6, nz+6) + fluxes = u.random_array_on_device(queue, nvars, ndim, nx+6, ny+6, nz+6) + metrics = u.random_array_on_device(queue, ndim, ndim, nx+6, ny+6, nz+6) + metric_jacobians = u.random_array_on_device(queue, nx+6, ny+6, nz+6) print("END ARRAY GEN") - flux_derivatives_dev = empty_array_on_device(queue, nvars, ndim, nx+6, ny+6, nz+6) + flux_derivatives_dev = u.empty_array_on_device(queue, nvars, ndim, nx+6, ny+6, nz+6) if write_code: - write_target_device_code(prg) + u.write_target_device_code(prg) allocator = pyopencl.tools.MemoryPool(pyopencl.tools.ImmediateAllocator(queue))