Skip to content

Commit

Permalink
Changes based on review
Browse files Browse the repository at this point in the history
Signed-off-by: Shreyas Khandekar <[email protected]>
  • Loading branch information
ShreyasKhandekar committed Oct 9, 2024
1 parent f008983 commit c49231d
Show file tree
Hide file tree
Showing 5 changed files with 7 additions and 8 deletions.
1 change: 0 additions & 1 deletion util/cron/test-gpu-ex-cuda-12.bash
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,6 @@ export CHPL_LLVM=bundled # CUDA 12 is only supported with bundled LLVM
export CHPL_COMM=none
export CHPL_LOCALE_MODEL=gpu
export CHPL_LAUNCHER_PARTITION=allgriz
export CHPL_TEST_GPU=true
export CHPL_GPU=nvidia # amd is also detected automatically

export CHPL_NIGHTLY_TEST_CONFIG_NAME="gpu-ex-cuda-12"
Expand Down
8 changes: 4 additions & 4 deletions util/cron/test-gpu-ex-cuda-12.interop.bash
Original file line number Diff line number Diff line change
Expand Up @@ -8,17 +8,17 @@ source $CWD/common-hpe-cray-ex.bash


# We need 12.4 for the stream test because the CUDA driver on pinoak
# only supports PTX for 12.4, untill the driver is updated, we need to
# only supports PTX for 12.4, until the driver is updated, we need to
# stick with 12.4 instead of 12.5
module load cuda/12.4 # default is CUDA 12
module load cuda/12.4 # default is CUDA 12.5

# We need cublas for the cublas interop test, but since we are using 12.4 above
# pinoak doesn't have the cublas library for 12.4, so we need to use the cublas
# from 12.5 (which is compatible across minor versions)
# This can be removed once we use CUDA 12.5
export CHPL_LIB_PATH="/opt/nvidia/hpc_sdk/Linux_x86_64/24.7/math_libs/lib64"
export CHPL_LIB_PATH="/opt/nvidia/hpc_sdk/Linux_x86_64/24.7/math_libs/lib64:$CHPL_LIB_PATH"

export CHPL_LLVM=bundled # CUDA 12 is only supported with bundled LLVM
export CHPL_LLVM=bundled # Using bundled LLVM since that's safer
export CHPL_TEST_GPU=true
export CHPL_LAUNCHER_PARTITION=allgriz
export CHPL_NIGHTLY_TEST_DIRS="gpu/interop/"
Expand Down
2 changes: 1 addition & 1 deletion util/cron/test-gpu-ex-cuda-12.specialization.bash
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ source $CWD/common-hpe-cray-ex.bash

module load cudatoolkit # default is CUDA 12

export CHPL_LLVM=bundled # CUDA 12 is only supported with bundled LLVM
export CHPL_LLVM=bundled # Using bundled LLVM since that's safer
export CHPL_COMM=none
export CHPL_LOCALE_MODEL=gpu
export CHPL_LAUNCHER_PARTITION=allgriz
Expand Down
2 changes: 1 addition & 1 deletion util/cron/test-perf.gpu-ex-cuda-12.bash
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ source $CWD/common-hpe-cray-ex.bash

module load cudatoolkit # default is CUDA 12

export CHPL_LLVM=bundled # CUDA 12 is only supported with bundled LLVM
export CHPL_LLVM=bundled # Using bundled LLVM since that's safer
export CHPL_COMM=none
export CHPL_LOCALE_MODEL=gpu
export CHPL_LAUNCHER_PARTITION=allgriz
Expand Down
2 changes: 1 addition & 1 deletion util/cron/test-perf.gpu-ex-cuda-12.um.bash
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ source $CWD/common-hpe-cray-ex.bash

module load cudatoolkit # default is CUDA 12

export CHPL_LLVM=bundled # CUDA 12 is only supported with bundled LLVM
export CHPL_LLVM=bundled # Using bundled LLVM since that's safer
export CHPL_COMM=none
export CHPL_LOCALE_MODEL=gpu
export CHPL_LAUNCHER_PARTITION=allgriz
Expand Down

0 comments on commit c49231d

Please sign in to comment.