Skip to content

Commit

Permalink
BatchScripts
Browse files Browse the repository at this point in the history
  • Loading branch information
OsKnoth committed Jul 12, 2024
1 parent ae47ca7 commit c73cd8a
Show file tree
Hide file tree
Showing 2 changed files with 32 additions and 0 deletions.
4 changes: 4 additions & 0 deletions BatchScripts/levante/runBaroWave.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
export JuliaDevice="GPU"
export JuliaGPU="CUDA"
export UCX_ERROR_SIGNALS=""
srun -n 1 gpu_wrapper.sh -n 1 -e "./Jobs/NHSphere/BaroWaveDrySphere"
28 changes: 28 additions & 0 deletions BatchScripts/lumi/runBaroWave.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
#SBATCH --account=project_465000863
#SBATCH --time=00:20:00
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=1
#SBATCH --gpus-per-node=1
#SBATCH --mem=0
#SBATCH --partition=dev-g
# dev-g max 56 cpus per task

#module load CrayEnv rocm

cat << EOF > select_gpu
#!/bin/bash
export JuliaDevice="GPU"
export JuliaGPU="AMD"
export MPICH_GPU_SUPPORT_ENABLED=1
export ROCR_VISIBLE_DEVICES=\$SLURM_LOCALID
exec \$*
EOF

chmod +x ./select_gpu

echo '####'

#CPU_BIND="map_cpu:49,57"
srun --cpu-bind=cores --unbuffered ./select_gpu ./Jobs/NHSphere/BaroWaveDrySphere

0 comments on commit c73cd8a

Please sign in to comment.