From caa9cad31ea2e577ac34f8695014f81f53c89e26 Mon Sep 17 00:00:00 2001 From: Rohan Potdar Date: Fri, 24 Jul 2026 16:14:03 -0500 Subject: [PATCH] [ROCm][Docker] Drop MORI_GPU_ARCHS so MoRI autodetects the device arch (#49737) Signed-off-by: Rohan Potdar Co-authored-by: Claude Opus 4.8 (1M context) --- docker/Dockerfile.rocm_base | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile.rocm_base b/docker/Dockerfile.rocm_base index 85f48ead469..a22bad04155 100644 --- a/docker/Dockerfile.rocm_base +++ b/docker/Dockerfile.rocm_base @@ -30,7 +30,7 @@ ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib: ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 ENV PYTORCH_ROCM_ARCH=${PYTORCH_ROCM_ARCH} ENV AITER_ROCM_ARCH=gfx942;gfx950 -ENV MORI_GPU_ARCHS=gfx942;gfx950 +# Note: Do not set MORI_GPU_ARCHS here, it is automatically inferred at runtime # Required for RCCL in ROCm7.1 ENV HSA_NO_SCRATCH_RECLAIM=1