AMD/ROCm: update AITER repo to ROCm/aiter (#3747)

docker/Dockerfile.rocm

@@ -17,8 +17,8 @@ ARG TRITON_REPO="https://github.com/ROCm/triton.git"
 ARG TRITON_COMMIT="improve_fa_decode_3.0.0"
 
 
-ARG ATER_REPO="https://github.com/HaiShaw/ater"
-ARG CK_COMMITS="fa05ae"
+ARG AITER_REPO="https://github.com/ROCm/aiter.git"
+ARG AITER_COMMIT="dev/testx"
 
 RUN git clone ${SGL_REPO} \
     && cd sglang \
@@ -53,8 +53,9 @@ RUN git clone ${TRITON_REPO} \
     && cd python \
     && python3 setup.py install
 
-RUN git clone ${ATER_REPO} \
-    && cd ater \
+RUN git clone ${AITER_REPO} \
+    && cd aiter \
+    && git checkout ${AITER_COMMIT} \
     && git submodule update --init --recursive \
     && PREBUILD_KERNELS=1 GPU_ARCHS=gfx942 python3 setup.py develop