Skip to content

Commit cbc95ff

Browse files
Update docker build convenience scripts to ROCm5.7 (#1543)
1 parent 17ea05e commit cbc95ff

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

libtorch/build_all_docker.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,6 @@ for cuda_version in 12.1 11.8; do
88
GPU_ARCH_TYPE=cuda GPU_ARCH_VERSION="${cuda_version}" "${TOPDIR}/libtorch/build_docker.sh"
99
done
1010

11-
for rocm_version in 5.5 5.6; do
11+
for rocm_version in 5.6 5.7; do
1212
GPU_ARCH_TYPE=rocm GPU_ARCH_VERSION="${rocm_version}" "${TOPDIR}/libtorch/build_docker.sh"
1313
done

manywheel/build_all_docker.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ for cuda_version in 12.1 11.8; do
1616
MANYLINUX_VERSION=2014 GPU_ARCH_TYPE=cuda GPU_ARCH_VERSION="${cuda_version}" "${TOPDIR}/manywheel/build_docker.sh"
1717
done
1818

19-
for rocm_version in 5.5 5.6; do
19+
for rocm_version in 5.6 5.7; do
2020
GPU_ARCH_TYPE=rocm GPU_ARCH_VERSION="${rocm_version}" "${TOPDIR}/manywheel/build_docker.sh"
2121
MANYLINUX_VERSION=2014 GPU_ARCH_TYPE=rocm GPU_ARCH_VERSION="${rocm_version}" "${TOPDIR}/manywheel/build_docker.sh"
2222
done

0 commit comments

Comments
 (0)