-
Notifications
You must be signed in to change notification settings - Fork 424
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add nightly testing for ROCm 5.4 (#23463)
The script is pretty similar to the CUDA test scripts on EX. Tested in jenkins manually. [Reviewed by @stonea]
- Loading branch information
Showing
2 changed files
with
26 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,23 @@ | ||
#!/usr/bin/env bash | ||
# | ||
# GPU native testing on a Cray EX (using none for CHPL_COMM) | ||
|
||
CWD=$(cd $(dirname ${BASH_SOURCE[0]}) ; pwd) | ||
source $CWD/common-native-gpu.bash | ||
source $CWD/common-hpe-cray-ex.bash | ||
|
||
module load rocm/5.4.3 # pin to rocm 5.4.3 | ||
|
||
export CHPL_COMM=none | ||
export CHPL_LLVM=bundled | ||
unset CHLP_LLVM_CONFIG # we need this to avoid warnings | ||
export CHPL_LOCALE_MODEL=gpu | ||
export CHPL_LAUNCHER_PARTITION=bardpeak # bardpeak is the default queue | ||
|
||
export CHPL_GPU=amd # also detected by default | ||
export CHPL_GPU_ARCH=gfx90a | ||
|
||
export CHPL_NIGHTLY_TEST_DIRS="gpu/native" | ||
|
||
export CHPL_NIGHTLY_TEST_CONFIG_NAME="gpu-ex-rocm-54" | ||
$CWD/nightly -cron ${nightly_args} |