diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index b9c481de5..30faed3dc 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -60,6 +60,14 @@ elif [[ "$OS_NAME" == *"Ubuntu"* ]]; then fi if [[ $ROCM_VERSION == "rocm3.7" ]]; then + TENSILE_LIBRARY_NAME=TensileLibrary.yaml +elif [[ $ROCM_VERSION == "rocm3.8" ]]; then + TENSILE_LIBRARY_NAME=TensileLibrary.dat +fi + +# NOTE: Some ROCm versions have identical dependencies. +# To avoid copy/paste mistakes, version condition branches are combined. +if [[ $ROCM_VERSION == "rocm3.7" || $ROCM_VERSION == "rocm3.8" ]]; then DEPS_LIST=( "/opt/rocm/miopen/lib/libMIOpen.so.1" "/opt/rocm/hip/lib/libamdhip64.so.3" @@ -109,7 +117,7 @@ DEPS_AUX_SRCLIST=( "/opt/rocm/rocblas/lib/library/TensileLibrary_gfx900.co" "/opt/rocm/rocblas/lib/library/TensileLibrary_gfx906.co" "/opt/rocm/rocblas/lib/library/TensileLibrary_gfx908.co" - "/opt/rocm/rocblas/lib/library/TensileLibrary.yaml" + "/opt/rocm/rocblas/lib/library/$TENSILE_LIBRARY_NAME" ) DEPS_AUX_DSTLIST=( @@ -121,7 +129,7 @@ DEPS_AUX_DSTLIST=( "lib/library/TensileLibrary_gfx900.co" "lib/library/TensileLibrary_gfx906.co" "lib/library/TensileLibrary_gfx908.co" - "lib/library/TensileLibrary.yaml" + "lib/library/$TENSILE_LIBRARY_NAME" ) else diff --git a/manywheel/deploy.sh b/manywheel/deploy.sh index 786e3d66a..e853028dd 100755 --- a/manywheel/deploy.sh +++ b/manywheel/deploy.sh @@ -2,7 +2,7 @@ set -eou pipefail -for rocm_version in 3.7; do +for rocm_version in 3.7 3.8; do ( set -x DOCKER_BUILDKIT=1 docker build \