From 59adcde08b91d92d9f2ce1d3799a3cc7d7c6fc33 Mon Sep 17 00:00:00 2001 From: Craig Warren Date: Wed, 31 Jul 2019 09:59:35 +0100 Subject: [PATCH] Updated checking of GPU field array shapes. --- gprMax/model_build_run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gprMax/model_build_run.py b/gprMax/model_build_run.py index b0f5bc94..70517b44 100644 --- a/gprMax/model_build_run.py +++ b/gprMax/model_build_run.py @@ -553,7 +553,7 @@ def solve_gpu(currentmodelrun, modelend, G): # Initialise arrays on GPU rxcoords_gpu, rxs_gpu = gpu_initialise_rx_arrays(G) # Prepare kernel and get kernel function - kernel_store_outputs = SourceModule(kernel_template_store_outputs.substitute(REAL=cudafloattype, NY_RXCOORDS=3, NX_RXS=6, NY_RXS=G.iterations, NZ_RXS=len(G.rxs), NX_FIELDS=(G.nx + 1, G.ny + 1, G.nz + 1), NY_FIELDS=(G.nx + 1, G.ny + 1, G.nz + 1), (G.nx + 1, G.ny + 1, G.nz + 1)), options=compiler_opts) + kernel_store_outputs = SourceModule(kernel_template_store_outputs.substitute(REAL=cudafloattype, NY_RXCOORDS=3, NX_RXS=6, NY_RXS=G.iterations, NZ_RXS=len(G.rxs), NX_FIELDS=(G.nx + 1, G.ny + 1, G.nz + 1), NY_FIELDS=(G.nx + 1, G.ny + 1, G.nz + 1), NZ_FIELDS=(G.nx + 1, G.ny + 1, G.nz + 1)), options=compiler_opts) store_outputs_gpu = kernel_store_outputs.get_function("store_outputs") # Sources - initialise arrays on GPU, prepare kernel and get kernel functions