diff --git a/check_binary.sh b/check_binary.sh index 103e1cacf..795d7e6ac 100755 --- a/check_binary.sh +++ b/check_binary.sh @@ -347,7 +347,7 @@ if [[ "$OSTYPE" == "msys" ]]; then fi # Test that CUDA builds are setup correctly -if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'rocm3.7' ]]; then +if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != *"rocm"* ]]; then if [[ "$PACKAGE_TYPE" == 'libtorch' ]]; then build_and_run_example_cpp check-torch-cuda else diff --git a/cron/update_s3_htmls.sh b/cron/update_s3_htmls.sh index e0d07578d..5bfe9428d 100755 --- a/cron/update_s3_htmls.sh +++ b/cron/update_s3_htmls.sh @@ -9,7 +9,7 @@ set -e # Upload for all CUDA/cpu versions if not given one to use if [[ -z "$CUDA_VERSIONS" ]]; then - export CUDA_VERSIONS=('cpu' 'cu92' 'cu100' 'cu101' 'cu102' 'cu110' 'rocm3.7') + export CUDA_VERSIONS=('cpu' 'cu92' 'cu100' 'cu101' 'cu102' 'cu110' 'rocm3.7' 'rocm3.8') fi if [[ -z "$HTML_NAME" ]]; then