From e9e1783cb6916c420cac6a12f8285a997acdebeb Mon Sep 17 00:00:00 2001 From: Jithun Nair Date: Tue, 19 Sep 2023 01:52:44 +0000 Subject: [PATCH] Update docker build convenience scripts to ROCm5.7 --- libtorch/build_all_docker.sh | 2 +- manywheel/build_all_docker.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/libtorch/build_all_docker.sh b/libtorch/build_all_docker.sh index e73e713b3..fb6bd975b 100755 --- a/libtorch/build_all_docker.sh +++ b/libtorch/build_all_docker.sh @@ -8,6 +8,6 @@ for cuda_version in 12.1 11.8; do GPU_ARCH_TYPE=cuda GPU_ARCH_VERSION="${cuda_version}" "${TOPDIR}/libtorch/build_docker.sh" done -for rocm_version in 5.5 5.6; do +for rocm_version in 5.6 5.7; do GPU_ARCH_TYPE=rocm GPU_ARCH_VERSION="${rocm_version}" "${TOPDIR}/libtorch/build_docker.sh" done diff --git a/manywheel/build_all_docker.sh b/manywheel/build_all_docker.sh index 2bd720f2f..2995e3be7 100644 --- a/manywheel/build_all_docker.sh +++ b/manywheel/build_all_docker.sh @@ -16,7 +16,7 @@ for cuda_version in 12.1 11.8; do MANYLINUX_VERSION=2014 GPU_ARCH_TYPE=cuda GPU_ARCH_VERSION="${cuda_version}" "${TOPDIR}/manywheel/build_docker.sh" done -for rocm_version in 5.5 5.6; do +for rocm_version in 5.6 5.7; do GPU_ARCH_TYPE=rocm GPU_ARCH_VERSION="${rocm_version}" "${TOPDIR}/manywheel/build_docker.sh" MANYLINUX_VERSION=2014 GPU_ARCH_TYPE=rocm GPU_ARCH_VERSION="${rocm_version}" "${TOPDIR}/manywheel/build_docker.sh" done