From 64563ed349d0153b122cac01a2f0e85ee7e35d52 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Martin=20Lilleeng=20S=C3=A6tra?= Date: Thu, 4 Mar 2021 17:52:02 +0100 Subject: [PATCH] Stupid multi-context multi-GPU working. Next: Cont w/imlp. of IC and inter-GPU comm. --- dgx-2-shmem-test.job | 2 +- shmemTesting.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dgx-2-shmem-test.job b/dgx-2-shmem-test.job index 392e4e3..57e8e88 100644 --- a/dgx-2-shmem-test.job +++ b/dgx-2-shmem-test.job @@ -3,7 +3,7 @@ #SBATCH -N 1 # number of nodes #SBATCH -n 1 # number of cores #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 -o slurm.%N.%j.out # STDOUT #SBATCH -e slurm.%N.%j.err # STDERR diff --git a/shmemTesting.py b/shmemTesting.py index 7518a38..8a6ec9d 100644 --- a/shmemTesting.py +++ b/shmemTesting.py @@ -94,7 +94,7 @@ for i in range(grid.ngpus): def genSim(**kwargs): sim = EE2D_KP07_dimsplit.EE2D_KP07_dimsplit(**kwargs) #sim = SHMEMSimulator.SHMEMSimulator(local_sim, grid) # implement SHMEMSimulator-support - sims.append(local_sim) + sims.append(sim) return sim outfile = Common.runSimulation(genSim, arguments, outfile, save_times, save_var_names)