Merge pull request #10 from JuliaParallel/initial

add LUMI configuration
JuliaParallel · Sep 17, 2024 · 792662f · 792662f
2 parents 8936fc4 + 5d19b1b
commit 792662f
Show file tree

Hide file tree

Showing 7 changed files with 73 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -97,7 +97,7 @@ export JUHPC_HDF5_HOME=$HDF5_DIR
 # Call JUHPC
 JUHPC_SETUP_INSTALLDIR=$SCRATCH/../julia/${HOSTNAME%%-*}/juhpc_setup
 JULIAUP_INSTALLDIR="\$SCRATCH/../julia/\$USER/\${HOSTNAME%%-*}/juliaup"
-VERSION="v0.1.0"
+VERSION="v0.1.1"
 wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/$VERSION/juhpc -O /tmp/juhpc
 bash -l /tmp/juhpc $JUHPC_SETUP_INSTALLDIR $JULIAUP_INSTALLDIR
 ```
@@ -124,7 +124,7 @@ export JUHPC_ADIOS2_HOME=$(spack -C $ENV_MOUNT/config location -i adios2)
 JUHPC_SETUP_INSTALLDIR=$ENV_MOUNT/juhpc_setup
 JULIAUP_INSTALLDIR="\$SCRATCH/../julia/\$USER/\${HOSTNAME%%-*}/juliaup"
 JUHPC_POST_INSTALL_JL=$ENV_EXTRA/uenv_view.jl
-VERSION="v0.1.0"
+VERSION="v0.1.1"
 wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/$VERSION/juhpc -O /tmp/juhpc
 bash -l /tmp/juhpc $JUHPC_SETUP_INSTALLDIR $JULIAUP_INSTALLDIR $JUHPC_POST_INSTALL_JL
 ```

diff --git a/configs/cscs/alps/gh200/craype_config b/configs/cscs/alps/gh200/craype_config
@@ -26,6 +26,6 @@ export JUHPC_HDF5_HOME=$HDF5_DIR
 # Call JUHPC
 JUHPC_SETUP_INSTALLDIR=$SCRATCH/${HOSTNAME%%-*}/juhpc_setup
 JULIAUP_INSTALLDIR="\$SCRATCH/\${HOSTNAME%%-*}/juliaup"
-VERSION="v0.1.0"
+VERSION="v0.1.1"
 wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/$VERSION/juhpc -O /tmp/juhpc
 bash -l /tmp/juhpc $JUHPC_SETUP_INSTALLDIR $JULIAUP_INSTALLDIR
diff --git a/configs/cscs/alps/mc/craype_config b/configs/cscs/alps/mc/craype_config
@@ -23,6 +23,6 @@ export JUHPC_HDF5_HOME=$HDF5_DIR
 # Call JUHPC
 JUHPC_SETUP_INSTALLDIR=$SCRATCH/${HOSTNAME%%-*}/juhpc_setup
 JULIAUP_INSTALLDIR="\$SCRATCH/\${HOSTNAME%%-*}/juliaup"
-VERSION="v0.1.0"
+VERSION="v0.1.1"
 wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/$VERSION/juhpc -O /tmp/juhpc
 bash -l /tmp/juhpc $JUHPC_SETUP_INSTALLDIR $JULIAUP_INSTALLDIR
diff --git a/configs/cscs/daint/gpu/craype_config b/configs/cscs/daint/gpu/craype_config
@@ -36,6 +36,6 @@ echo 'using Preferences
 # Call JUHPC
 JUHPC_SETUP_INSTALLDIR=$SCRATCH/../julia/${HOSTNAME%%[0-9]*}-gpu/juhpc_setup
 JULIAUP_INSTALLDIR="\$SCRATCH/../julia/\$USER/\${HOSTNAME%%[0-9]*}-gpu/juliaup"
-VERSION="v0.1.0"
+VERSION="v0.1.1"
 wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/$VERSION/juhpc -O /tmp/juhpc
 bash -l /tmp/juhpc $JUHPC_SETUP_INSTALLDIR $JULIAUP_INSTALLDIR $JUHPC_POST_INSTALL_JL
diff --git a/configs/cscs/daint/mc/craype_config b/configs/cscs/daint/mc/craype_config
@@ -23,6 +23,6 @@ export JUHPC_HDF5_HOME=$HDF5_DIR
 # Call JUHPC
 JUHPC_SETUP_INSTALLDIR=$SCRATCH/../julia/${HOSTNAME%%[0-9]*}-mc/juhpc_setup
 JULIAUP_INSTALLDIR="\$SCRATCH/../julia/\$USER/\${HOSTNAME%%[0-9]*}-mc/juliaup"
-VERSION="v0.1.0"
+VERSION="v0.1.1"
 wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/$VERSION/juhpc -O /tmp/juhpc
 bash -l /tmp/juhpc $JUHPC_SETUP_INSTALLDIR $JULIAUP_INSTALLDIR
diff --git a/configs/eurohpc/lumi/mi250x/craype_config b/configs/eurohpc/lumi/mi250x/craype_config
@@ -0,0 +1,29 @@
+#!/bin/bash
+
+# Author: Samuel Omlin, CSCS (omlins)
+#
+# Description: Definition of site specific variables and call of JUHPC.
+# Site: LUMI, EuroHPC JU
+# Base: craype (LUMI-flavored)
+
+# Load required modules (including correct CPU and GPU target modules)
+module load LUMI
+module load partition/G         # loads CPU and GPU target modules (craype-x86-trento, craype-accel-amd-gfx90a)
+module load cpeGNU              # LUMI-wrapper for PrgEnv-gnu
+module load rocm
+module load cray-hdf5-parallel
+module list
+
+# Environment variables for HPC key packages that require system libraries that require system libraries (MPI.jl, CUDA.jl, HDF5.jl and ADIOS2.jl)
+export JUHPC_ROCM_HOME=$ROCM_PATH
+export JUHPC_MPI_VENDOR="cray"
+export JUHPC_MPI_EXEC="srun"
+export JUHPC_HDF5_HOME=$HDF5_DIR
+
+
+# Call JUHPC
+JUHPC_SETUP_INSTALLDIR=$SCRATCH/${HOSTNAME%%[0-9]*}/juhpc_setup  #SCRATCH is assumed to be defined in ~/.bashrc, e.g., SCRATCH=/scratch/project_465000105/$USER
+JULIAUP_INSTALLDIR="\$SCRATCH/\${HOSTNAME%%[0-9]*}/juliaup"
+VERSION="v0.1.1"
+wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/$VERSION/juhpc -O /tmp/juhpc
+bash -l /tmp/juhpc $JUHPC_SETUP_INSTALLDIR $JULIAUP_INSTALLDIR
diff --git a/configs/eurohpc/lumi/mi250x/test_craype_config b/configs/eurohpc/lumi/mi250x/test_craype_config
@@ -0,0 +1,38 @@
+#!/bin/bash
+
+# Variable set in craype_config
+JUHPC_SETUP_INSTALLDIR=$SCRATCH/${HOSTNAME%%[0-9]*}/juhpc_setup
+
+# Load required modules (including correct CPU and GPU target modules)
+module load LUMI
+module load partition/G         # loads CPU and GPU target modules (craype-x86-trento, craype-accel-amd-gfx90a)
+module load cpeGNU              # LUMI-wrapper for PrgEnv-gnu
+module load rocm
+module load cray-hdf5-parallel
+module list
+
+# Activate the HPC setup environment variables
+. $JUHPC_SETUP_INSTALLDIR/activate
+
+# Call juliaup to install juliaup and latest julia on scratch
+juliaup
+
+# Call juliaup to see its options
+juliaup
+
+# Call julia Pkg 
+julia -e 'using Pkg; Pkg.status()'
+
+# Add AMDGPU.jl
+julia -e 'using Pkg; Pkg.add("AMDGPU"); using AMDGPU; AMDGPU.versioninfo()'
+
+# Add MPI.jl
+julia -e 'using Pkg; Pkg.add("MPI"); using MPI; MPI.versioninfo()'
+
+# Add HDF5.jl
+julia -e 'using Pkg; Pkg.add("HDF5"); using HDF5; @show HDF5.has_parallel()'
+
+# Test AMDGPU-aware MPI
+julia -e 'using Pkg; Pkg.add(["ImplicitGlobalGrid", "ParallelStencil"]);'
+cd ~/rocmaware
+MPICH_GPU_SUPPORT_ENABLED=1 IGG_ROCMAWARE_MPI=1 srun --time=00:09:00 -pdev-g -Aproject_465000557 -N2 -n2 --gpus-per-node=8 julia -O3 --check-bounds=no diffusion3D.jl