Skip to content

Commit dd94d30

Browse files
committed
Use source code mirror for CI
Signed-off-by: ZhanruiSunCh <184402041+ZhanruiSunCh@users.noreply.github.com>
1 parent ee941ac commit dd94d30

File tree

4 files changed

+27
-10
lines changed

4 files changed

+27
-10
lines changed

docker/Dockerfile.multi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ RUN GITHUB_MIRROR=${GITHUB_MIRROR} bash ./install_triton.sh && rm install_triton
107107
RUN if [ -f /etc/redhat-release ]; then \
108108
echo "Rocky8 detected, skipping mooncake installation"; \
109109
else \
110-
bash ./install_mooncake.sh; \
110+
GITHUB_MIRROR=${GITHUB_MIRROR} bash ./install_mooncake.sh; \
111111
fi && rm install_mooncake.sh
112112

113113
FROM ${DEVEL_IMAGE} AS wheel

docker/common/install_mooncake.sh

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,12 @@
11
#!/bin/bash
22
set -ex
33

4+
GITHUB_URL="https://github.com"
5+
if [ -n "${GITHUB_MIRROR}" ]; then
6+
GITHUB_URL=${GITHUB_MIRROR}
7+
fi
8+
49
MOONCAKE_VERSION="v0.3.6.post1"
5-
MOONCAKE_REPO="https://github.com/kvcache-ai/Mooncake.git"
610
MOONCAKE_INSTALL_PATH="/usr/local/Mooncake"
711

812
apt-get update
@@ -27,8 +31,10 @@ apt-get install -y --no-install-recommends \
2731

2832
mkdir -p /third-party-source
2933

30-
git clone --depth 1 https://github.com/alibaba/yalantinglibs.git
31-
tar -czf /third-party-source/yalantinglibs.tar.gz yalantinglibs
34+
YALANTINGLIBS_VERSION="v1.0.0"
35+
curl -L ${GITHUB_URL}/alibaba/yalantinglibs/archive/refs/tags/${YALANTINGLIBS_VERSION}.tar.gz -o yalantinglibs-${YALANTINGLIBS_VERSION}.tar.gz
36+
tar -xzf yalantinglibs-${YALANTINGLIBS_VERSION}.tar.gz
37+
mv yalantinglibs-${YALANTINGLIBS_VERSION} yalantinglibs
3238
cd yalantinglibs
3339
mkdir build && cd build
3440
cmake .. -DBUILD_EXAMPLES=OFF -DBUILD_BENCHMARK=OFF -DBUILD_UNIT_TESTS=OFF
@@ -37,8 +43,9 @@ make install
3743
cd ../..
3844
rm -rf yalantinglibs
3945

40-
git clone --depth 1 -b ${MOONCAKE_VERSION} ${MOONCAKE_REPO}
41-
tar -czf /third-party-source/Mooncake-${MOONCAKE_VERSION}.tar.gz Mooncake
46+
curl -L ${GITHUB_URL}/kvcache-ai/Mooncake/archive/refs/tags/${MOONCAKE_VERSION}.tar.gz -o Mooncake-${MOONCAKE_VERSION}.tar.gz
47+
tar -xzf Mooncake-${MOONCAKE_VERSION}.tar.gz
48+
mv Mooncake-${MOONCAKE_VERSION} Mooncake
4249
cd Mooncake
4350
git submodule update --init --recursive --depth 1
4451
mkdir build && cd build

docker/common/install_nixl.sh

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,12 @@
22
set -ex
33

44
GITHUB_URL="https://github.com"
5+
if [ -n "${GITHUB_MIRROR}" ]; then
6+
GITHUB_URL=${GITHUB_MIRROR}
7+
fi
58
UCX_INSTALL_PATH="/usr/local/ucx/"
69
CUDA_PATH="/usr/local/cuda"
710
NIXL_VERSION="0.5.0"
8-
NIXL_REPO="https://github.com/ai-dynamo/nixl.git"
911
OLD_LD_LIBRARY_PATH=$LD_LIBRARY_PATH
1012

1113
ARCH_NAME="x86_64-linux-gnu"
@@ -20,7 +22,9 @@ if [ -n "${GITHUB_MIRROR}" ]; then
2022
fi
2123
pip3 install --no-cache-dir meson ninja pybind11
2224

23-
git clone --depth 1 -b ${NIXL_VERSION} ${NIXL_REPO}
25+
curl -L ${GITHUB_URL}/ai-dynamo/nixl/archive/refs/tags/${NIXL_VERSION}.tar.gz -o nixl-${NIXL_VERSION}.tar.gz
26+
tar -xzf nixl-${NIXL_VERSION}.tar.gz
27+
mv nixl-${NIXL_VERSION} nixl
2428
cd nixl
2529

2630
CUDA_SO_PATH=$(find "/usr/local" -name "libcuda.so.1" 2>/dev/null | head -n1)

docker/common/install_ucx.sh

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,19 @@
11
#!/bin/bash
22
set -ex
33

4+
GITHUB_URL="https://github.com"
5+
if [ -n "${GITHUB_MIRROR}" ]; then
6+
GITHUB_URL=${GITHUB_MIRROR}
7+
fi
8+
49
UCX_VERSION="v1.19.x"
510
UCX_INSTALL_PATH="/usr/local/ucx/"
611
CUDA_PATH="/usr/local/cuda"
7-
UCX_REPO="https://github.com/openucx/ucx.git"
812

913
rm -rf ${UCX_INSTALL_PATH}
10-
git clone --depth 1 -b ${UCX_VERSION} ${UCX_REPO}
14+
curl -L ${GITHUB_URL}/openucx/ucx/archive/refs/tags/${UCX_VERSION}.tar.gz -o ucx-${UCX_VERSION}.tar.gz
15+
tar -xzf ucx-${UCX_VERSION}.tar.gz
16+
mv ucx-${UCX_VERSION} ucx
1117
cd ucx
1218
./autogen.sh
1319
./contrib/configure-release \

0 commit comments

Comments
 (0)