File tree Expand file tree Collapse file tree 4 files changed +27
-10
lines changed Expand file tree Collapse file tree 4 files changed +27
-10
lines changed Original file line number Diff line number Diff line change @@ -107,7 +107,7 @@ RUN GITHUB_MIRROR=${GITHUB_MIRROR} bash ./install_triton.sh && rm install_triton
107107RUN 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
113113FROM ${DEVEL_IMAGE} AS wheel
Original file line number Diff line number Diff line change 11#! /bin/bash
22set -ex
33
4+ GITHUB_URL=" https://github.com"
5+ if [ -n " ${GITHUB_MIRROR} " ]; then
6+ GITHUB_URL=${GITHUB_MIRROR}
7+ fi
8+
49MOONCAKE_VERSION=" v0.3.6.post1"
5- MOONCAKE_REPO=" https://github.com/kvcache-ai/Mooncake.git"
610MOONCAKE_INSTALL_PATH=" /usr/local/Mooncake"
711
812apt-get update
@@ -27,8 +31,10 @@ apt-get install -y --no-install-recommends \
2731
2832mkdir -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
3238cd yalantinglibs
3339mkdir build && cd build
3440cmake .. -DBUILD_EXAMPLES=OFF -DBUILD_BENCHMARK=OFF -DBUILD_UNIT_TESTS=OFF
@@ -37,8 +43,9 @@ make install
3743cd ../..
3844rm -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
4249cd Mooncake
4350git submodule update --init --recursive --depth 1
4451mkdir build && cd build
Original file line number Diff line number Diff line change 22set -ex
33
44GITHUB_URL=" https://github.com"
5+ if [ -n " ${GITHUB_MIRROR} " ]; then
6+ GITHUB_URL=${GITHUB_MIRROR}
7+ fi
58UCX_INSTALL_PATH=" /usr/local/ucx/"
69CUDA_PATH=" /usr/local/cuda"
710NIXL_VERSION=" 0.5.0"
8- NIXL_REPO=" https://github.com/ai-dynamo/nixl.git"
911OLD_LD_LIBRARY_PATH=$LD_LIBRARY_PATH
1012
1113ARCH_NAME=" x86_64-linux-gnu"
@@ -20,7 +22,9 @@ if [ -n "${GITHUB_MIRROR}" ]; then
2022fi
2123pip3 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
2428cd nixl
2529
2630CUDA_SO_PATH=$( find " /usr/local" -name " libcuda.so.1" 2> /dev/null | head -n1)
Original file line number Diff line number Diff line change 11#! /bin/bash
22set -ex
33
4+ GITHUB_URL=" https://github.com"
5+ if [ -n " ${GITHUB_MIRROR} " ]; then
6+ GITHUB_URL=${GITHUB_MIRROR}
7+ fi
8+
49UCX_VERSION=" v1.19.x"
510UCX_INSTALL_PATH=" /usr/local/ucx/"
611CUDA_PATH=" /usr/local/cuda"
7- UCX_REPO=" https://github.com/openucx/ucx.git"
812
913rm -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
1117cd ucx
1218./autogen.sh
1319./contrib/configure-release \
You can’t perform that action at this time.
0 commit comments