From 5ef67cee160b979fbd880cfd7b90e8b97fbc15fc Mon Sep 17 00:00:00 2001 From: "Vladimir (Vova) Vagaytsev" <10628074+vvagaytsev@users.noreply.github.com> Date: Thu, 16 Apr 2026 03:40:05 +0200 Subject: [PATCH] [AMD] Fix `aiter` import failure in ROCm Docker images (#22363) --- docker/rocm.Dockerfile | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/rocm.Dockerfile b/docker/rocm.Dockerfile index abc2f8615..2107482d9 100644 --- a/docker/rocm.Dockerfile +++ b/docker/rocm.Dockerfile @@ -184,12 +184,12 @@ RUN cd aiter \ && echo "[AITER] GPU_ARCH=${GPU_ARCH}" \ && if [ "$BUILD_AITER_ALL" = "1" ] && [ "$BUILD_LLVM" = "1" ]; then \ sh -c "HIP_CLANG_PATH=/sgl-workspace/llvm-project/build/bin/ PREBUILD_KERNELS=1 GPU_ARCHS=$GPU_ARCH_LIST python setup.py build_ext --inplace" \ - && sh -c "HIP_CLANG_PATH=/sgl-workspace/llvm-project/build/bin/ GPU_ARCHS=$GPU_ARCH_LIST pip install -e ."; \ + && sh -c "HIP_CLANG_PATH=/sgl-workspace/llvm-project/build/bin/ GPU_ARCHS=$GPU_ARCH_LIST pip install --config-settings editable_mode=compat -e ."; \ elif [ "$BUILD_AITER_ALL" = "1" ]; then \ sh -c "PREBUILD_KERNELS=1 GPU_ARCHS=$GPU_ARCH_LIST python setup.py build_ext --inplace" \ - && sh -c "GPU_ARCHS=$GPU_ARCH_LIST pip install -e ."; \ + && sh -c "GPU_ARCHS=$GPU_ARCH_LIST pip install --config-settings editable_mode=compat -e ."; \ else \ - sh -c "GPU_ARCHS=$GPU_ARCH_LIST pip install -e ."; \ + sh -c "GPU_ARCHS=$GPU_ARCH_LIST pip install --config-settings editable_mode=compat -e ."; \ fi \ && echo "export PYTHONPATH=/sgl-workspace/aiter:\${PYTHONPATH}" >> /etc/bash.bashrc