From 430711c76a52dad5d1165ad2d6699abb5683702b Mon Sep 17 00:00:00 2001 From: Jeff Daily Date: Wed, 2 Dec 2020 17:16:20 -0800 Subject: [PATCH] add rocm 3.10 manywheel support --- cron/update_s3_htmls.sh | 2 +- manywheel/build_rocm.sh | 2 +- manywheel/deploy.sh | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/cron/update_s3_htmls.sh b/cron/update_s3_htmls.sh index 7164213ed..07d13b3ae 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' 'rocm3.8' 'rocm3.9') + export CUDA_VERSIONS=('cpu' 'cu92' 'cu100' 'cu101' 'cu102' 'cu110' 'rocm3.7' 'rocm3.8' 'rocm3.9' 'rocm3.10') fi if [[ -z "$HTML_NAME" ]]; then diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 9002829ab..680a9a19c 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -75,7 +75,7 @@ fi # NOTE: Some ROCm versions have identical dependencies, or very close deps. # To avoid copy/paste mistakes, version condition branches are combined. -if [[ $ROCM_VERSION == "rocm3.7" || $ROCM_VERSION == "rocm3.8" || $ROCM_VERSION == "rocm3.9" ]]; then +if [[ $ROCM_VERSION == "rocm3.7" || $ROCM_VERSION == "rocm3.8" || $ROCM_VERSION == "rocm3.9" || $ROCM_VERSION == "rocm3.10" ]]; then DEPS_LIST=( "/opt/rocm/miopen/lib/libMIOpen.so.1" diff --git a/manywheel/deploy.sh b/manywheel/deploy.sh index ce4aac81f..132ed96c7 100755 --- a/manywheel/deploy.sh +++ b/manywheel/deploy.sh @@ -2,7 +2,7 @@ set -eou pipefail -for rocm_version in 3.7 3.8 3.9; do +for rocm_version in 3.7 3.8 3.9 3.10; do ( set -x DOCKER_BUILDKIT=1 docker build \