fix(ascend): preserve validated triton runtime

Author: AlexMa616

.github/configs/ascend.yml

@@ -61,7 +61,7 @@ image_build:
       dockerfile: docker/ascend/Dockerfile.inference
       image: flagos-dev/flagscale-inference
       test_roles: [inference]
-      base_image: harbor.baai.ac.cn/flagscale/vllm-plugin-fl@sha256:5062ac23b8b8247e5b4faa5d2c61b7b8e82e46a9ae7c55fe15c5cdbe3533f4b7
+      base_image: harbor.baai.ac.cn/flagos-dev/flagscale:manual-20260812-ascend-dev-inference@sha256:e5b8e3b2c37584f980dffa1c08649b44e9edd2415522e123502e8d82a7854ef1
       build_args: {}
       source_refs:
         VLLM_PLUGIN_FL_REF: vllm_plugin_fl

docker/ascend/Dockerfile.inference

@@ -12,9 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# The immutable base supplies the matching Ascend vLLM runtime. FlagScale
-# overlays the resolved vllm-plugin-FL and FlagGems revisions.
-ARG BASE_IMAGE=harbor.baai.ac.cn/flagscale/vllm-plugin-fl@sha256:5062ac23b8b8247e5b4faa5d2c61b7b8e82e46a9ae7c55fe15c5cdbe3533f4b7
+# The immutable base owns the matching Ascend vLLM, FlagTree and Triton runtime.
+# FlagScale only overlays the resolved vllm-plugin-FL and FlagGems revisions.
+ARG BASE_IMAGE=harbor.baai.ac.cn/flagos-dev/flagscale:manual-20260812-ascend-dev-inference@sha256:e5b8e3b2c37584f980dffa1c08649b44e9edd2415522e123502e8d82a7854ef1
 
 FROM ${BASE_IMAGE} AS dev
 

requirements/ascend/serve.txt

@@ -12,11 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# Ascend serve-specific dependencies. The immutable inference image owns the
-# Python and vLLM dependency closure, so this file must not include base.txt or
-# resolve transitive packages that can replace the vendor-tested runtime.
+# The immutable inference image owns the complete Python, vLLM, FlagTree and
+# Triton dependency closure. Installing FlagTree over that image would replace
+# files in the shared triton namespace and invalidate the tested compiler stack.
 
-# FlagTree for Ascend (extra index so PyPI remains available for build deps)
-# [--extra-index-url https://resource.flagos.net/repository/flagos-pypi-hosted/simple]
+# Keep the ascend-serve packaging extra explicit. The image installer validates
+# this base-owned package instead of reinstalling it.
 # [--no-deps]
 flagtree==0.4.0+ascend3.2

tools/install/ascend/install_serve.sh

@@ -28,6 +28,7 @@ RETRY_COUNT="${FLAGSCALE_RETRY_COUNT:-3}"
 FLAGSCALE_HOME="${FLAGSCALE_HOME:-/opt/flagscale}"
 FLAGSCALE_DEPS="${FLAGSCALE_DEPS:-$FLAGSCALE_HOME/deps}"
 REQ_FILE="$PROJECT_ROOT/requirements/ascend/serve.txt"
+EXPECTED_FLAGTREE_VERSION="0.4.0+ascend3.2"
 FLAGGEMS_REPO="${FLAGSCALE_FLAGGEMS_REPO:-https://github.com/flagos-ai/FlagGems.git}"
 FLAGGEMS_REF="${FLAGSCALE_FLAGGEMS_REF:-v5.3.0}"
 VLLM_PLUGIN_REPO="${FLAGSCALE_VLLM_PLUGIN_REPO:-https://github.com/flagos-ai/vllm-plugin-FL.git}"
@@ -45,10 +46,20 @@ done
 install_pip() {
     if is_phase_enabled task; then
         [ ! -f "$REQ_FILE" ] && { log_info "serve.txt not found"; return 0; }
-        set_step "Installing serve requirements"
-        retry_pip_install -d $DEBUG "$REQ_FILE" "$RETRY_COUNT" || return 1
-        [ "$DEBUG" = true ] || python -m pip show flagtree >/dev/null || return 1
-        log_success "Serve requirements installed"
+        set_step "Validating base-owned Ascend serve runtime"
+        if [ "$DEBUG" != true ]; then
+            EXPECTED_FLAGTREE_VERSION="$EXPECTED_FLAGTREE_VERSION" python - <<'PY' || return 1
+import importlib.metadata as metadata
+import os
+
+expected = os.environ["EXPECTED_FLAGTREE_VERSION"]
+actual = metadata.version("flagtree")
+assert actual == expected, f"FlagTree version mismatch: expected {expected}, found {actual}"
+
+import triton  # noqa: F401, E402
+PY
+        fi
+        log_success "Ascend serve runtime validated"
     else
         local pkgs=$(get_pip_deps_for_requirements "$REQ_FILE")
         [ -z "$pkgs" ] && return 0