smyalygames/FiniteVolumeGPU_HIP
1
0
Fork 0
mirror of https://github.com/smyalygames/FiniteVolumeGPU_HIP.git synced 2025-05-18 06:24:11 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Simulator.py with hip-python

Browse Source
This commit is contained in:
Hicham Agueny 2024-02-26 12:39:56 +01:00
parent 655913021c
commit ca9171a5bf
1 changed files with 24 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
GPUSimulators/Simulator.py
View File

@ -25,16 +25,15 @@ import numpy as np
import logging
from enum import IntEnum
import pycuda.compiler as cuda_compiler
import pycuda.gpuarray
import pycuda.driver as cuda
#import pycuda.compiler as cuda_compiler
#import pycuda.gpuarray
#import pycuda.driver as cuda
from hip import hip, hiprtc
from GPUSimulators import Common
class BoundaryCondition(object):
"""
Class for holding boundary conditions for global boundaries
@ -101,13 +100,17 @@ class BoundaryCondition(object):
class BaseSimulator(object):
def hip_check(call_result):
err = call_result[0]
result = call_result[1:]
if len(result) == 1:
result = result[0]
if isinstance(err, hip.hipError_t) and err != hip.hipError_t.hipSuccess:
raise RuntimeError(str(err))
return result
def __init__(self,
context,
nx, ny,
@ -161,9 +164,11 @@ class BaseSimulator(object):
)
#Create a CUDA stream
self.stream = cuda.Stream()
self.internal_stream = cuda.Stream()
#self.stream = cuda.Stream()
#self.internal_stream = cuda.Stream()
self.stream = hip_check(hip.hipStreamCreate())
self.internal_stream = hip_check(hip.hipStreamCreate())
#Keep track of simulation time and number of timesteps
self.t = 0.0
self.nt = 0
@ -231,7 +236,9 @@ class BaseSimulator(object):
return self.getOutput().download(self.stream, variables)
def synchronize(self):
self.stream.synchronize()
#self.stream.synchronize()
#Synchronize the stream to ensure operations in the stream is complete
hip_check(hip.hipStreamSynchronize(self.stream))
def simTime(self):
return self.t
@ -268,15 +275,6 @@ class BaseSimulator(object):
def stepOrderToCodedInt(step, order):
"""
Helper function which packs the step and order into a single integer
@ -285,4 +283,4 @@ def stepOrderToCodedInt(step, order):
#print("Step: {0:032b}".format(step))
#print("Order: {0:032b}".format(order))
#print("Mix: {0:032b}".format(step_order))
return np.int32(step_order)
return np.int32(step_order)