|
| 1 | +#!/bin/bash |
| 2 | + |
| 3 | +# Author: Samuel Omlin, CSCS (omlins) |
| 4 | +# |
| 5 | +# Description: Definition of site specific variables and call of JUHPC. |
| 6 | +# Site: LUMI, EuroHPC JU |
| 7 | +# Base: craype |
| 8 | + |
| 9 | + |
| 10 | +# module load LUMI/22.08 |
| 11 | +# module load partition/G |
| 12 | +# module load rocm/5.3.3 |
| 13 | +# module load cray-hdf5-parallel |
| 14 | + |
| 15 | +# module load CrayEnv |
| 16 | +# module switch PrgEnv-cray PrgEnv-gnu #PrgEnv-amd |
| 17 | +# module load rocm craype-accel-nvidia90 |
| 18 | +# module load cray-hdf5-parallel |
| 19 | +# module list |
| 20 | + |
| 21 | +# Load required modules (including correct CPU and GPU target modules) |
| 22 | +module load LUMI |
| 23 | +module load partition/G # loads CPU and GPU target modules (craype-x86-trento, craype-accel-amd-gfx90a) |
| 24 | +module load cpeGNU # LUMI-wrapper for PrgEnv-gnu |
| 25 | +module load rocm |
| 26 | +module load cray-hdf5-parallel |
| 27 | +module list |
| 28 | + |
| 29 | +# Environment variables for HPC key packages that require system libraries that require system libraries (MPI.jl, CUDA.jl, HDF5.jl and ADIOS2.jl) |
| 30 | +export JUHPC_ROCM_HOME=$ROCM_PATH |
| 31 | +export JUHPC_MPI_VENDOR="cray" |
| 32 | +export JUHPC_MPI_EXEC="srun" |
| 33 | +export JUHPC_HDF5_HOME=$HDF5_DIR |
| 34 | + |
| 35 | + |
| 36 | +# Call JUHPC |
| 37 | +JUHPC_SETUP_INSTALLDIR=$SCRATCH/${HOSTNAME%%[0-9]*}/juhpc_setup #SCRATCH is assumed to be defined in ~/.bashrc, e.g., SCRATCH=/scratch/project_465000105/$USER |
| 38 | +JULIAUP_INSTALLDIR="\$SCRATCH/\${HOSTNAME%%[0-9]*}/juliaup" |
| 39 | +VERSION="v0.1.0" |
| 40 | +#wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/$VERSION/juhpc -O /tmp/juhpc |
| 41 | +wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/main/juhpc -O /tmp/juhpc |
| 42 | +bash -l /tmp/juhpc $JUHPC_SETUP_INSTALLDIR $JULIAUP_INSTALLDIR |
0 commit comments