Skip to content

Commit ad21d0b

Browse files
committed
add daint/gpu/craype_config_no_cudaaware
1 parent 53db395 commit ad21d0b

File tree

1 file changed

+28
-0
lines changed

1 file changed

+28
-0
lines changed
Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
#!/bin/bash
2+
3+
# Author: Samuel Omlin, CSCS (omlins)
4+
#
5+
# Description: Definition of site specific variables and call of JUHPC.
6+
# Site: Piz Daint:gpu, Swiss National Supercomputing Centre (CSCS)
7+
# Base: craype (not using CUDA-aware MPI because system CUDA is outdated as daint is to be decommissioned soon)
8+
9+
10+
# Load required modules, including correct CPU and GPU target modules (NOTE: the same modules should be loaded when running julia - JUHPC can be used in a module build recipe...)
11+
module load daint-gpu
12+
module switch PrgEnv-cray PrgEnv-gnu
13+
module load cray-hdf5-parallel
14+
module list
15+
16+
17+
# Environment variables for HPC key packages that require system libraries that require system libraries (MPI.jl, CUDA.jl, AMDGPU.jl, HDF5.jl and ADIOS2.jl)
18+
export JUHPC_MPI_HOME=$MPICH_DIR
19+
export JUHPC_MPI_EXEC="srun -C gpu"
20+
export JUHPC_HDF5_HOME=$HDF5_DIR
21+
22+
23+
# Call JUHPC
24+
JUHPC_SETUP_INSTALLDIR=$SCRATCH/../julia/${HOSTNAME%%[0-9]*}-gpu-nocudaaware/juhpc_setup
25+
JULIAUP_INSTALLDIR="\$SCRATCH/../julia/\$USER/\${HOSTNAME%%[0-9]*}-gpu-nocudaaware/juliaup"
26+
VERSION="v0.2.0"
27+
wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/$VERSION/juhpc -O /tmp/juhpc
28+
bash -l /tmp/juhpc $JUHPC_SETUP_INSTALLDIR $JULIAUP_INSTALLDIR

0 commit comments

Comments
 (0)