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

Removed nvtx and MPI options for job script

Browse Source
This commit is contained in:
Martin Lilleeng Sætra
2022-05-25 14:52:44 +02:00
parent a588948e77
commit a260230002
2 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
GPUSimulators/MPISimulator.py
View File

@@ -27,7 +27,7 @@ from mpi4py import MPI
import time
import pycuda.driver as cuda
import nvtx
#import nvtx
@@ -314,14 +314,14 @@ class MPISimulator(Simulator.BaseSimulator):
def substep(self, dt, step_number):
nvtx.mark("substep start", color="yellow")
#nvtx.mark("substep start", color="yellow")
self.profiling_data_mpi["start"]["t_mpi_step"] += time.time()
nvtx.mark("substep external", color="blue")
#nvtx.mark("substep external", color="blue")
self.sim.substep(dt, step_number, external=True, internal=False) # only "internal ghost cells"
nvtx.mark("substep internal", color="red")
#nvtx.mark("substep internal", color="red")
self.sim.substep(dt, step_number, internal=True, external=False) # "internal ghost cells" excluded
#nvtx.mark("substep full", color="blue")
@@ -331,7 +331,7 @@ class MPISimulator(Simulator.BaseSimulator):
self.profiling_data_mpi["end"]["t_mpi_step"] += time.time()
nvtx.mark("exchange", color="blue")
#nvtx.mark("exchange", color="blue")
self.full_exchange()
#nvtx.mark("download", color="blue")
@@ -343,10 +343,10 @@ class MPISimulator(Simulator.BaseSimulator):
#nvtx.mark("upload", color="blue")
#self.upload_for_exchange(self.sim.u0)
nvtx.mark("sync start", color="blue")
#nvtx.mark("sync start", color="blue")
self.sim.stream.synchronize()
self.sim.internal_stream.synchronize()
nvtx.mark("sync end", color="blue")
#nvtx.mark("sync end", color="blue")
self.profiling_data_mpi["n_time_steps"] += 1