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

Stupid multi-context multi-GPU working.

Browse Source 
Next: Cont w/imlp. of IC and inter-GPU comm.
This commit is contained in:
Martin Lilleeng Sætra 2021-03-04 17:52:02 +01:00
parent aade33caa5
commit 64563ed349
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dgx-2-shmem-test.job
View File

@ -3,7 +3,7 @@
#SBATCH -N 1 # number of nodes #SBATCH -N 1 # number of nodes
#SBATCH -n 1 # number of cores #SBATCH -n 1 # number of cores
#SBATCH -w g001 # DGX-2 node #SBATCH -w g001 # DGX-2 node
#SBATCH --gres=gpu:4 # number of V100's #SBATCH --gres=gpu:2 # number of V100's
#SBATCH -t 0-00:10 # time (D-HH:MM) #SBATCH -t 0-00:10 # time (D-HH:MM)
#SBATCH -o slurm.%N.%j.out # STDOUT #SBATCH -o slurm.%N.%j.out # STDOUT
#SBATCH -e slurm.%N.%j.err # STDERR #SBATCH -e slurm.%N.%j.err # STDERR

2
shmemTesting.py
View File

@ -94,7 +94,7 @@ for i in range(grid.ngpus):
def genSim(**kwargs): def genSim(**kwargs):
sim = EE2D_KP07_dimsplit.EE2D_KP07_dimsplit(**kwargs) sim = EE2D_KP07_dimsplit.EE2D_KP07_dimsplit(**kwargs)
#sim = SHMEMSimulator.SHMEMSimulator(local_sim, grid) # implement SHMEMSimulator-support #sim = SHMEMSimulator.SHMEMSimulator(local_sim, grid) # implement SHMEMSimulator-support
sims.append(local_sim) sims.append(sim)
return sim return sim
outfile = Common.runSimulation(genSim, arguments, outfile, save_times, save_var_names) outfile = Common.runSimulation(genSim, arguments, outfile, save_times, save_var_names)