smyalygames/FiniteVolumeGPU
1
0
Fork 0
mirror of https://github.com/smyalygames/FiniteVolumeGPU.git synced 2025-10-31 20:17:41 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(common): allow for empty cpu_data to be parsed

Browse Source
This commit is contained in:
Anthony Berg 2025-07-01 01:30:20 +02:00
parent a59c0f073d
commit d112dc98bd
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
GPUSimulators/common/arrays/arkawa2d.py
View File

@ -1,7 +1,5 @@
import logging
# from .typing import array2d
class BaseArakawaA2D(object):
"""

12
GPUSimulators/common/arrays/hip/array2d.py
View File

@ -21,10 +21,14 @@ class HIPArray2D(BaseArray2D):
self.data_h = np.zeros(self.shape, self.dtype)
self.num_bytes = self.data_h.size * self.data_h.itemsize
self.data_d = hip_check(hip.hipMalloc(self.num_bytes)).configure(
self.data = hip_check(hip.hipMalloc(self.num_bytes)).configure(
typestr=np.finfo(self.dtype).dtype.name, shape=self.shape
)
# If there is no data to append, just leave this array as allocated
if cpu_data is None:
return
# Create a copy object from host to device
x = (self.shape[0] - cpu_data.shape[1]) // 2
y = (self.shape[1] - cpu_data.shape[0]) // 2
@ -33,7 +37,7 @@ class HIPArray2D(BaseArray2D):
def __del__(self, *args):
# self.logger.debug("Buffer <%s> [%dx%d]: Releasing ", int(self.data.gpudata), self.nx, self.ny)
hip_check(hip.hipFree(self.data_d))
hip_check(hip.hipFree(self.data))
def download(self, stream, cpu_data=None, asynch=False, extent=None):
"""
@ -59,7 +63,7 @@ class HIPArray2D(BaseArray2D):
hip_check(hip.hipStreamSynchronize(stream))
hip_check(
hip.hipMemcpyAsync(self.data_d, cpu_data, self.num_bytes, hip.hipMemcpyKind.hipMemcpyDeviceToHost, stream))
hip.hipMemcpyAsync(self.data, cpu_data, self.num_bytes, hip.hipMemcpyKind.hipMemcpyDeviceToHost, stream))
return cpu_data
@ -76,5 +80,5 @@ class HIPArray2D(BaseArray2D):
# TODO implement non-async to test if it actually works - avoid errors
# Create a copy object from device to host
hip_check(hip.hipMemcpyAsync(self.data_d, self.data_h, self.num_bytes, hip.hipMemcpyKind.hipMemcpyHostToDevice,
hip_check(hip.hipMemcpyAsync(self.data, self.data_h, self.num_bytes, hip.hipMemcpyKind.hipMemcpyHostToDevice,
stream))