smyalygames/FiniteVolumeGPU
1
0
Fork 0
mirror of https://github.com/smyalygames/FiniteVolumeGPU.git synced 2026-01-14 15:48:43 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Refactoring

Browse Source
This commit is contained in:
André R. Brodtkorb
2018-11-01 20:43:28 +01:00
parent 2b899d1c80
commit d9eb72d78c
6 changed files with 220 additions and 221 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
GPUSimulators/Common.py
View File

@@ -92,7 +92,7 @@ class CudaArray2D:
#self.logger.debug("Allocating [%dx%d] buffer", self.nx, self.ny)
#Should perhaps use pycuda.driver.mem_alloc_data.pitch() here
self.data = pycuda.gpuarray.empty((ny_halo, nx_halo), dtype)
self.data = pycuda.gpuarray.zeros((ny_halo, nx_halo), dtype)
#If we don't have any data, just allocate and return
if cpu_data is None:
@@ -310,5 +310,12 @@ class ArakawaA2D:
stream.synchronize()
return cpu_variables
"""
Checks that data is still sane
"""
def check(self):
for i, gpu_variable in enumerate(self.gpu_variables):
var_sum = pycuda.gpuarray.sum(gpu_variable.data).get()
self.logger.debug("Data %d with size [%d x %d] has sum %f", i, gpu_variable.nx, gpu_variable.ny, var_sum)
assert np.isnan(var_sum) == False, "Data contains NaN values!"