Skip to content

Commit 4f43dae

Browse files
author
Doug Lehr
committed
Fix ARCH arguments
1 parent 1b1e7cd commit 4f43dae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docker/Dockerfile.rocm_base

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ FROM ${BASE_IMAGE} AS base
1818
ENV PATH=/opt/rocm/llvm/bin:$PATH
1919
ENV ROCM_PATH=/opt/rocm
2020
ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:
21-
ARG PYTORCH_ROCM_ARCH=gfx942;gfx950;
21+
ARG PYTORCH_ROCM_ARCH=gfx942;gfx950
2222
ENV PYTORCH_ROCM_ARCH=${PYTORCH_ROCM_ARCH}
2323
# ENV AITER_ROCM_ARCH=gfx942;gfx950
2424

0 commit comments

Comments
 (0)