diff --git a/docker/rocm.Dockerfile b/docker/rocm.Dockerfile index e3a9e8717..de91105ec 100644 --- a/docker/rocm.Dockerfile +++ b/docker/rocm.Dockerfile @@ -67,7 +67,7 @@ ARG LLVM_BRANCH="MainOpSelV2" ARG LLVM_COMMIT="6520ace8227ffe2728148d5f3b9872a870b0a560" ARG MOONCAKE_REPO="https://github.com/kvcache-ai/Mooncake.git" -ARG MOONCAKE_COMMIT="dcdf1c784b40aa6975a8ed89fe26321b028e40e8" +ARG MOONCAKE_COMMIT="7c22adb0d2c1616e3cfefb2dc6bf01c23c3df768" ARG TILELANG_REPO="https://github.com/HaiShaw/tilelang.git" ARG TILELANG_BRANCH="dsv32-mi35x" @@ -156,7 +156,7 @@ RUN if [ "$BUILD_MOONCAKE" = "1" ]; then \ rm go1.22.2.linux-amd64.tar.gz && \ mkdir -p build && \ cd build && \ - cmake .. -DUSE_ETCD=ON && \ + cmake .. -DUSE_HIP=ON -DUSE_ETCD=ON && \ make -j "$(nproc)" && make install; \ fi