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

create time Event with hip

Browse Source
This commit is contained in:
Hicham Agueny 2024-01-17 17:19:07 +01:00
parent 28ab54a86e
commit b90035c902
1 changed files with 35 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
GPUSimulators/Autotuner.py
View File

@ -27,11 +27,25 @@ import logging
from socket import gethostname from socket import gethostname
#import pycuda.driver as cuda #import pycuda.driver as cuda
import pyhip as cuda from hip import hip,hiprtc
from GPUSimulators import Common, Simulator, CudaContext from GPUSimulators import Common, Simulator, CudaContext
class Autotuner: class Autotuner:
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))
elif (
isinstance(err, hiprtc.hiprtcResult)
and err != hiprtc.hiprtcResult.HIPRTC_SUCCESS
):
raise RuntimeError(str(err))
return result
def __init__(self, def __init__(self,
nx=2048, ny=2048, nx=2048, ny=2048,
block_widths=range(8, 32, 1), block_widths=range(8, 32, 1),
@ -184,24 +198,35 @@ class Autotuner:
return np.nan return np.nan
#Create timer events #Create timer events
start = cuda.Event() #start = cuda.Event()
end = cuda.Event() #end = cuda.Event()
stream = hip_check(hip.hipStreamCreate())
start = hip_check(hip.hipEventCreate())
end = hip_check(hip.hipEventCreate())
#Warmup #Warmup
for i in range(warmup_timesteps): for i in range(warmup_timesteps):
sim.stepEuler(sim.dt) sim.stepEuler(sim.dt)
#Run simulation with timer #Run simulation with timer
start.record(sim.stream) #start.record(sim.stream)
#start recording
hip_check(hip.hipEventRecord(start, stream))
for i in range(timesteps): for i in range(timesteps):
sim.stepEuler(sim.dt) sim.stepEuler(sim.dt)
end.record(sim.stream) #end.record(sim.stream)
#stop recording and synchronize
hip_check(hip.hipEventRecord(end, stream))
#Synchronize end event #Synchronize end event
end.synchronize() #end.synchronize()
hip_check(hip.hipEventSynchronize(end))
#Compute megacells #Compute megacells
gpu_elapsed = end.time_since(start)*1.0e-3 #gpu_elapsed = end.time_since(start)*1.0e-3
gpu_elapsed = hip_check(hip.hipEventElapsedTime(start, end))
megacells = (sim.nx*sim.ny*timesteps / (1000*1000)) / gpu_elapsed megacells = (sim.nx*sim.ny*timesteps / (1000*1000)) / gpu_elapsed
#Sanity check solution #Sanity check solution