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)