Skip to content

Commit 53d1e1a

Browse files
committed
add gh200 configuration test
1 parent 05452ab commit 53d1e1a

File tree

1 file changed

+36
-0
lines changed

1 file changed

+36
-0
lines changed
Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
#!/bin/bash
2+
3+
# Variable set in craype_config
4+
JUHPC_SETUP_INSTALLDIR=$SCRATCH/${HOSTNAME%%-*}/juhpc_setup # HPC setup installation environment variables must be expanded during installation.
5+
6+
# Load required modules (including correct CPU and GPU target modules)
7+
module load cray
8+
module switch PrgEnv-cray PrgEnv-gnu
9+
module load cudatoolkit craype-accel-nvidia90
10+
module load cray-hdf5-parallel
11+
module list
12+
13+
# Activate the HPC setup environment variables
14+
. $JUHPC_SETUP_INSTALLDIR/activate
15+
16+
# Call juliaup to install juliaup and latest julia on scratch
17+
juliaup
18+
19+
# Call juliaup to see its options
20+
juliaup
21+
22+
# Call julia Pkg
23+
julia -e 'using Pkg; Pkg.status()'
24+
25+
# Add CUDA.jl
26+
julia -e 'using Pkg; Pkg.add("CUDA"); using CUDA; CUDA.versioninfo()'
27+
28+
# Add MPI.jl
29+
julia -e 'using Pkg; Pkg.add("MPI"); using MPI; MPI.versioninfo()'
30+
31+
# Add HDF5.jl
32+
julia -e 'using Pkg; Pkg.add("HDF5"); using HDF5; @show HDF5.has_parallel()'
33+
34+
# Test CUDA-aware MPI
35+
cd ~/cudaaware
36+
MPICH_GPU_SUPPORT_ENABLED=1 srun -Acsstaff -C'gpu' -N2 -n2 julia cudaaware.jl

0 commit comments

Comments
 (0)