Adding Navi3 target (#327)

gfx1030;gfx1100;gfx1101
ROCm · Dec 13, 2024 · 9ddae90 · 9ddae90
1 parent 9932d97
commit 9ddae90
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile.rocm b/Dockerfile.rocm
@@ -24,8 +24,8 @@ USER root
 ARG BASE_IMAGE
 ARG COMMON_WORKDIR
 # Used as ARCHes for all Pytorch supported ARCH
-ARG ARG_PYTORCH_ROCM_ARCH
-ENV PYTORCH_ROCM_ARCH=${ARG_PYTORCH_ROCM_ARCH:-${PYTORCH_ROCM_ARCH}}
+ARG ARG_PYTORCH_ROCM_ARCH="gfx90a;gfx942;gfx1030;gfx1100;gfx1101"
+ENV PYTORCH_ROCM_ARCH=${ARG_PYTORCH_ROCM_ARCH}
 
 # Install some basic utilities
 RUN apt-get update -q -y && apt-get install -q -y python3 python3-pip