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

feat(GPUSimulator): add the extra variables introduced in SWWECommon for all other algorithms

Browse Source
This commit is contained in:
Anthony Berg
2025-02-18 10:21:12 +01:00
parent 71cc5628c8
commit aa54d21f0a
16 changed files with 255 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
GPUSimulators/KP07_dimsplit.py
View File

@@ -46,7 +46,9 @@ class KP07_dimsplit(Simulator.BaseSimulator):
theta=1.3,
cfl_scale=0.9,
boundary_conditions=BoundaryCondition(),
block_width=16, block_height=16):
block_width=16, block_height=16,
dt: float=None,
compile_opts: list[str]=[]):
"""
Initialization routine
@@ -60,6 +62,7 @@ class KP07_dimsplit(Simulator.BaseSimulator):
dy: Grid cell spacing along y-axis (20 000 m)
dt: Size of each timestep (90 s)
g: Gravitational accelleration (9.81 m/s^2)
compile_opts: Pass a list of nvcc compiler options
"""
# Call super constructor
@@ -83,11 +86,11 @@ class KP07_dimsplit(Simulator.BaseSimulator):
},
compile_args={
'no_extern_c': True,
'options': ["--use_fast_math"],
'options': ["--use_fast_math"] + compile_opts,
},
jit_compile_args={})
self.kernel = module.get_function("KP07DimsplitKernel")
self.kernel.prepare("iifffffiiPiPiPiPiPiPiP")
self.kernel.prepare("iifffffiiPiPiPiPiPiPiPiiii")
#Create data by uploading to device
self.u0 = Common.ArakawaA2D(self.stream,
@@ -99,10 +102,15 @@ class KP07_dimsplit(Simulator.BaseSimulator):
self.gc_x, self.gc_y,
[None, None, None])
self.cfl_data = gpuarray.GPUArray(self.grid_size, dtype=np.float32)
dt_x = np.min(self.dx / (np.abs(hu0/h0) + np.sqrt(g*h0)))
dt_y = np.min(self.dy / (np.abs(hv0/h0) + np.sqrt(g*h0)))
dt = min(dt_x, dt_y)
self.cfl_data.fill(dt, stream=self.stream)
if dt == None:
dt_x = np.min(self.dx / (np.abs(hu0/h0) + np.sqrt(g*h0)))
dt_y = np.min(self.dy / (np.abs(hv0/h0) + np.sqrt(g*h0)))
self.dt = min(dt_x, dt_y)
else:
self.dt = dt
self.cfl_data.fill(self.dt, stream=self.stream)
def substep(self, dt, step_number):
self.substepDimsplit(dt*0.5, step_number)
@@ -121,7 +129,9 @@ class KP07_dimsplit(Simulator.BaseSimulator):
self.u1[0].data.gpudata, self.u1[0].data.strides[0],
self.u1[1].data.gpudata, self.u1[1].data.strides[0],
self.u1[2].data.gpudata, self.u1[2].data.strides[0],
self.cfl_data.gpudata)
self.cfl_data.gpudata,
0, 0,
self.nx, self.ny)
self.u0, self.u1 = self.u1, self.u0
def getOutput(self):