Skip to content

Commit

Permalink
Merge remote-tracking branch 'origin/develop' into mccoords_abstraction
Browse files Browse the repository at this point in the history
  • Loading branch information
ye-luo committed Jan 29, 2022
2 parents f504324 + 28db9e7 commit 5115498
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion tests/test_automation/github-actions/ci/run_step.sh
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,15 @@ case "$1" in
;;
esac

if [[ "${GH_JOBNAME}" =~ (-CUDA) ]]
then
echo "Set CUDACXX CMake environment variable to nvcc standard location"
export CUDACXX=/usr/local/cuda/bin/nvcc

# Make current environment variables available to subsequent steps
echo "CUDACXX=/usr/local/cuda/bin/nvcc" >> $GITHUB_ENV
fi

# Sanitizer
case "${GH_JOBNAME}" in
*"ASan"*)
Expand Down Expand Up @@ -267,7 +276,8 @@ case "$1" in
echo "Enabling OpenMPI oversubscription"
export OMPI_MCA_rmaps_base_oversubscribe=1
export OMPI_MCA_hwloc_base_binding_policy=none
if [[ "$HOST_NAME" =~ (sulfur) ]]

if [[ "$HOST_NAME" =~ (sulfur) || "$HOST_NAME" =~ (nitrogen) ]]
then
echo "Set the management layer to ucx"
export OMPI_MCA_pml=ucx
Expand Down

0 comments on commit 5115498

Please sign in to comment.