Skip to content

Commit 7cf8722

Browse files
authored
[r2.20] CP PR 3022 Adjust gpu archs list format and environment variable (#3150)
1 parent 6f9203f commit 7cf8722

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.manylinux_2_28

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,9 @@ COPY setup.packages.rocm.el8.sh setup.packages.rocm.el8.sh
1717
COPY builder.packages.rocm.el8.txt builder.packages.rocm.el8.txt
1818
RUN /setup.packages.rocm.el8.sh /builder.packages.rocm.el8.txt
1919

20-
ARG GPU_DEVICE_TARGETS="gfx908 gfx90a gfx942 gfx1030 gfx1100 gfx1101 gfx1102 gfx1200 gfx1201"
21-
ENV GPU_DEVICE_TARGETS=${GPU_DEVICE_TARGETS}
20+
ARG GPU_DEVICE_TARGETS="gfx908,gfx90a,gfx942,gfx950,gfx1030,gfx1100,gfx1101,gfx1102,gfx1200,gfx1201"
21+
ARG GPU_DEVICE_TARGETS=${GPU_DEVICE_TARGETS}
22+
ENV TF_ROCM_AMDGPU_TARGETS=${GPU_DEVICE_TARGETS}
2223

2324
# Install ROCM
2425
ARG ROCM_VERSION

0 commit comments

Comments
 (0)