[ROCm][Docker] Drop MORI_GPU_ARCHS so MoRI autodetects the device arch (#49737)
Signed-off-by: Rohan Potdar <rohan.potdar@amd.com> Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
Rohan Potdar committed
Jul 24, 2026 at 16:14 UTC
caa9cad31ea2e577ac34f8695014f81f53c89e26
1 file changed
+1
-1
docker/Dockerfile.rocm_base
+1
-1
@@ -30,7 +30,7 @@ ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:
30
ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151
31
ENV PYTORCH_ROCM_ARCH=${PYTORCH_ROCM_ARCH}
32
ENV AITER_ROCM_ARCH=gfx942;gfx950
33
-ENV MORI_GPU_ARCHS=gfx942;gfx950
33
+# Note: Do not set MORI_GPU_ARCHS here, it is automatically inferred at runtime
34
35
# Required for RCCL in ROCm7.1
36
ENV HSA_NO_SCRATCH_RECLAIM=1