smyalygames/FiniteVolumeGPU
1
0
Fork 0
mirror of https://github.com/smyalygames/FiniteVolumeGPU.git synced 2025-09-15 03:32:17 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(common): bad import of hip_check

Browse Source
This commit is contained in:
Anthony Berg 2025-07-01 01:14:06 +02:00
parent 86821d659c
commit a59c0f073d
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,10 +1,8 @@
import ctypes
import numpy as np import numpy as np
from hip import hip, hipblas from hip import hip
from GPUSimulators.common import hip_check from ...hip_check import hip_check
from GPUSimulators.common.arrays.array2d import BaseArray2D from ..array2d import BaseArray2D
class HIPArray2D(BaseArray2D): class HIPArray2D(BaseArray2D):
@ -24,7 +22,7 @@ class HIPArray2D(BaseArray2D):
self.num_bytes = self.data_h.size * self.data_h.itemsize self.num_bytes = self.data_h.size * self.data_h.itemsize
self.data_d = hip_check(hip.hipMalloc(self.num_bytes)).configure( self.data_d = hip_check(hip.hipMalloc(self.num_bytes)).configure(
typestr=np.finfo(dtype).dtype.name, shape=self.shape typestr=np.finfo(self.dtype).dtype.name, shape=self.shape
) )
# Create a copy object from host to device # Create a copy object from host to device