smyalygames/FiniteVolumeGPU
1
0
Fork 0
mirror of https://github.com/smyalygames/FiniteVolumeGPU.git synced 2025-11-29 17:28:03 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Profiling code for DGX-2. Added fixed dt.

Browse Source
This commit is contained in:
Martin Lilleeng Sætra
2022-04-08 15:09:09 +02:00
parent 80d84e0489
commit d83d620512
5 changed files with 92 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
GPUSimulators/Common.py
View File

@@ -89,7 +89,7 @@ def toJson(in_dict, compressed=True):
out_dict[key] = value
return json.dumps(out_dict)
def runSimulation(simulator, simulator_args, outfile, save_times, save_var_names=[]):
def runSimulation(simulator, simulator_args, outfile, save_times, save_var_names=[], dt=None):
"""
Runs a simulation, and stores output in netcdf file. Stores the times given in
save_times, and saves all of the variables in list save_var_names. Elements in
@@ -176,7 +176,7 @@ def runSimulation(simulator, simulator_args, outfile, save_times, save_var_names
#Simulate
if (t_step > 0.0):
sim.simulate(t_step)
sim.simulate(t_step, dt)
profiling_data_sim_runner["end"]["t_step"] += time.time()