From e3d8385bc2f3276fb21fba69afe88807bcddf00e Mon Sep 17 00:00:00 2001 From: Rong Rong Date: Thu, 24 Sep 2020 12:22:16 -0700 Subject: [PATCH] [hotfix] follow up changes to rocm3.8 --- check_binary.sh | 2 +- cron/update_s3_htmls.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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