Skip to content

Commit

Permalink
Add GPU ugni testing on Cray XC (#24945)
Browse files Browse the repository at this point in the history
This adds a nightly testing config for gpu testing using ugni comm layer
on xc

[Reviewed by @e-kayrakli ]
  • Loading branch information
ShreyasKhandekar authored Apr 30, 2024
2 parents e68366a + 7493cff commit cb81008
Showing 1 changed file with 28 additions and 0 deletions.
28 changes: 28 additions & 0 deletions util/cron/test-gpu-xc-cuda.ugni.bash
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
#!/usr/bin/env bash
#
# Build Arkouda on a cray-xc with GPU locale model

CWD=$(cd $(dirname ${BASH_SOURCE[0]}) ; pwd)

source $CWD/common-native-gpu.bash

# Setup for GPU:
module load cudatoolkit
export CHPL_TARGET_COMPILER=llvm
export CHPL_LLVM=bundled
export CHPL_TARGET_CPU=native
export CHPL_LAUNCHER_CONSTRAINT=BW18
export CHPL_LAUNCHER="slurm-srun"

# setup for XC perf (ugni, gnu, 28-core broadwell)
module unload $(module -t list 2>&1 | grep PrgEnv-)
module load PrgEnv-gnu
module unload $(module -t list 2>&1 | grep craype-hugepages)
module load craype-hugepages16M
module unload perftools-base
module unload atp
module load craype-x86-cascadelake


export CHPL_NIGHTLY_TEST_CONFIG_NAME="gpu-xc-cuda.ugni"
$CWD/nightly -cron ${nightly_args}

0 comments on commit cb81008

Please sign in to comment.