diff --git a/docker/rocm.Dockerfile b/docker/rocm.Dockerfile index 78f3ac79f..712d7f5bd 100644 --- a/docker/rocm.Dockerfile +++ b/docker/rocm.Dockerfile @@ -199,7 +199,7 @@ RUN set -eux; \ # it will be removed when server launched issue is fixed by aiter RUN cd aiter \ && echo "[AITER] GPU_ARCH=${GPU_ARCH}" \ - && sed -i '/c1 = torch.empty((M, D, S1 + S3), dtype=dtype, device=x.device)/i\ config = dict(config)' aiter/ops/triton/gemm/fused/fused_gemm_afp4wfp4_split_cat.py; \ + && sed -i '/c1 = torch.empty((M, D, S1 + S3), dtype=dtype, device=x.device)/i\ config = dict(config)' aiter/ops/triton/gemm/fused/fused_gemm_afp4wfp4_split_cat.py \ && 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 develop"; \ elif [ "$BUILD_AITER_ALL" = "1" ]; then \