Skip to content

Commit e4db178

Browse files
committed
Mooncake need update submodule and build, can't use mirror
Signed-off-by: ZhanruiSunCh <[email protected]>
1 parent 7a52e3f commit e4db178

File tree

1 file changed

+5
-15
lines changed

1 file changed

+5
-15
lines changed

docker/common/install_mooncake.sh

Lines changed: 5 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,8 @@ if [ -n "${GITHUB_MIRROR}" ]; then
66
GITHUB_URL=${GITHUB_MIRROR}
77
fi
88

9-
MOONCAKE_VERSION="0.3.6.post1"
9+
MOONCAKE_VERSION="v0.3.6.post1"
10+
MOONCAKE_REPO="https://github.com/kvcache-ai/Mooncake.git"
1011
MOONCAKE_INSTALL_PATH="/usr/local/Mooncake"
1112

1213
apt-get update
@@ -45,21 +46,10 @@ make install
4546
cd ../..
4647
rm -rf yalantinglibs
4748

48-
curl -L ${GITHUB_URL}/kvcache-ai/Mooncake/archive/refs/tags/v${MOONCAKE_VERSION}.tar.gz -o Mooncake-${MOONCAKE_VERSION}.tar.gz
49-
tar -xzf Mooncake-${MOONCAKE_VERSION}.tar.gz
50-
# Auto-detect extracted directory name
51-
MOONCAKE_DIR=$(tar -tzf Mooncake-${MOONCAKE_VERSION}.tar.gz | head -1 | cut -f1 -d"/")
52-
mv ${MOONCAKE_DIR} Mooncake
49+
git clone --depth 1 -b ${MOONCAKE_VERSION} ${MOONCAKE_REPO}
50+
tar -czf /third-party-source/Mooncake-${MOONCAKE_VERSION}.tar.gz Mooncake
5351
cd Mooncake
54-
55-
# Manually download pybind11 submodule (since tarball doesn't include submodules)
56-
PYBIND11_VERSION="v2.13.6"
57-
mkdir -p extern
58-
curl -L ${GITHUB_URL}/pybind/pybind11/archive/refs/tags/${PYBIND11_VERSION}.tar.gz -o pybind11.tar.gz
59-
tar -xzf pybind11.tar.gz
60-
PYBIND11_DIR=$(tar -tzf pybind11.tar.gz | head -1 | cut -f1 -d"/")
61-
mv ${PYBIND11_DIR} extern/pybind11
62-
rm pybind11.tar.gz
52+
git submodule update --init --recursive --depth 1
6353

6454
mkdir build && cd build
6555
cmake .. -DUSE_CUDA=ON -DBUILD_SHARED_LIBS=ON -DCMAKE_INSTALL_PREFIX=${MOONCAKE_INSTALL_PATH}

0 commit comments

Comments
 (0)