Skip to content

Commit 3daa37e

Browse files
committed
Updated to use ROCm5.4.2
1 parent 69d11a6 commit 3daa37e

File tree

5 files changed

+6
-6
lines changed

5 files changed

+6
-6
lines changed

.github/workflows/build-libtorch-images.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ jobs:
4747
runs-on: ubuntu-18.04
4848
strategy:
4949
matrix:
50-
rocm_version: ["5.3", "5.4.1"]
50+
rocm_version: ["5.3", "5.4.2"]
5151
env:
5252
GPU_ARCH_TYPE: rocm
5353
GPU_ARCH_VERSION: ${{ matrix.rocm_version }}

.github/workflows/build-manywheel-images.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ jobs:
4949
runs-on: ubuntu-18.04
5050
strategy:
5151
matrix:
52-
rocm_version: ["5.3", "5.4.1"]
52+
rocm_version: ["5.3", "5.4.2"]
5353
env:
5454
GPU_ARCH_TYPE: rocm
5555
GPU_ARCH_VERSION: ${{ matrix.rocm_version }}

libtorch/Dockerfile

+2-2
Original file line numberDiff line numberDiff line change
@@ -74,8 +74,8 @@ RUN ROCM_VERSION=5.3 bash ./install_rocm.sh && rm install_rocm.sh
7474
RUN bash ./install_rocm_drm.sh && rm install_rocm_drm.sh
7575
RUN bash ./install_rocm_magma.sh && rm install_rocm_magma.sh
7676

77-
FROM rocm as rocm5.4.1
78-
RUN ROCM_VERSION=5.4.1 bash ./install_rocm.sh && rm install_rocm.sh
77+
FROM rocm as rocm5.4.2
78+
RUN ROCM_VERSION=5.4.2 bash ./install_rocm.sh && rm install_rocm.sh
7979
RUN bash ./install_rocm_drm.sh && rm install_rocm_drm.sh
8080
RUN bash ./install_rocm_magma.sh && rm install_rocm_magma.sh
8181

libtorch/build_all_docker.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,6 @@ for cuda_version in 11.8 11.7 11.6; 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.3 5.4.1; do
11+
for rocm_version in 5.3 5.4.2; do
1212
GPU_ARCH_TYPE=rocm GPU_ARCH_VERSION="${rocm_version}" "${TOPDIR}/libtorch/build_docker.sh"
1313
done

manywheel/build_all_docker.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ for cuda_version in 11.7 11.6; do
1414
MANYLINUX_VERSION=2014 GPU_ARCH_TYPE=cuda GPU_ARCH_VERSION="${cuda_version}" "${TOPDIR}/manywheel/build_docker.sh"
1515
done
1616

17-
for rocm_version in 5.3 5.4.1; do
17+
for rocm_version in 5.3 5.4.2; do
1818
GPU_ARCH_TYPE=rocm GPU_ARCH_VERSION="${rocm_version}" "${TOPDIR}/manywheel/build_docker.sh"
1919
MANYLINUX_VERSION=2014 GPU_ARCH_TYPE=rocm GPU_ARCH_VERSION="${rocm_version}" "${TOPDIR}/manywheel/build_docker.sh"
2020
done

0 commit comments

Comments
 (0)