@@ -103,20 +103,7 @@ RUN pip_install() { pip install "$@" || pip install --index-url https://pypi.org
103103 cd /root/MindSpeed && git checkout core_r0.16.0 && pip install -r requirements.txt && pip install -e . && git checkout e4772499 && \
104104 cd /root/Megatron-LM && git checkout core_v0.16.1 && pip install -e . --no-build-isolation && \
105105 cd /root/Megatron-Bridge && git checkout v0.3.1 && \
106- cd /root/MindSpeed-Ops && git checkout 33ac80f7 && \
107- # GitCode's release CDN may require a proxy even when gitcode.com is in NO_PROXY.
108- makeself_url=https://gitcode.com/cann-src-third-party/makeself/releases/download/release-2.5.0-patch1.0/makeself-release-2.5.0-patch1.tar.gz && \
109- makeself_redirect=$(curl -fsS -o /dev/null -w '%{redirect_url}' "$makeself_url") && \
110- curl -fSL --noproxy '' "${makeself_redirect:-$makeself_url}" -o /tmp/makeself.tar.gz && \
111- echo 'bfa730a5763cdb267904a130e02b2e48e464986909c0733ff1c96495f620369a /tmp/makeself.tar.gz' | sha256sum -c - && \
112- mkdir -p mindspeed_ops/csrc/third_party/makeself && \
113- tar -xzf /tmp/makeself.tar.gz --strip-components=1 -C mindspeed_ops/csrc/third_party/makeself && \
114- rm /tmp/makeself.tar.gz && \
115- # This upstream archive is byte-identical to CANN's GitCode mirror.
116- curl -fSL https://github.com/nlohmann/json/releases/download/v3.11.3/include.zip -o /tmp/json.zip && \
117- echo 'a22461d13119ac5c78f205d3df1db13403e58ce1bb1794edc9313677313f4a9d /tmp/json.zip' | sha256sum -c - && \
118- unzip -q /tmp/json.zip -d mindspeed_ops/csrc/third_party/json && rm /tmp/json.zip && \
119- pip install -e . --no-build-isolation --no-deps && \
106+ cd /root/MindSpeed-Ops && git checkout 33ac80f7 && pip install -e . --no-build-isolation --no-deps && \
120107 cd /root/MindSpeed-Bridge && git checkout v0.3.1 && pip install -r requirements.txt && pip install -e . --no-deps
121108
122109# Keep Relax code changes out of the dependency cache.
@@ -164,20 +151,6 @@ RUN git clone https://github.com/sgl-project/sgl-kernel-npu /root/sgl-kernel-npu
164151# Install AscendC FLA
165152RUN git clone https://github.com/flashserve/flash-linear-attention-npu.git /root/flash-linear-attention-npu && \
166153 cd /root/flash-linear-attention-npu && git checkout v26.6.0 && \
167- # Reuse verified CANN dependencies and prefetch FLA's remaining pinned archives.
168- # Resolve GitCode redirects before using the proxy for its release CDN.
169- mkdir -p third_party/pkg && \
170- cp -a /root/MindSpeed-Ops/mindspeed_ops/csrc/third_party/{makeself,json} third_party/ && \
171- for dependency in \
172- 'eigen/releases/download/5.0.0-h0.trunk/eigen-5.0.0.tar.gz 93f7f0462988b934e632a9fba58af55192ffceae38e8f46233f4f62cb1e79370' \
173- 'protobuf/releases/download/v25.1/protobuf-25.1.tar.gz 9bd87b8280ef720d3240514f884e56a712f2218f0d693b48050c836028940a42' \
174- 'abseil-cpp/releases/download/20230802.1/abseil-cpp-20230802.1.tar.gz 987ce98f02eefbaf930d6e38ab16aa05737234d7afbab2d5c4ea7adbe50c28ed'; do \
175- read -r path checksum <<< "$dependency"; \
176- url="https://gitcode.com/cann-src-third-party/$path"; filename="${path##*/}"; \
177- redirect=$(curl -fsS -o /dev/null -w '%{redirect_url}' "$url") && \
178- curl -fSL --noproxy '' "${redirect:-$url}" -o "third_party/pkg/$filename" && \
179- printf '%s %s\n' "$checksum" "third_party/pkg/$filename" | sha256sum -c - || exit 1; \
180- done && \
181154 apt update && apt-get install -y --no-install-recommends gawk && \
182155 source /usr/local/Ascend/ascend-toolkit/set_env.sh && \
183156 source /usr/local/Ascend/nnal/atb/set_env.sh && \
0 commit comments