From d1c2a1de080cb0aa55b95085c5cafef2f9bdc93d Mon Sep 17 00:00:00 2001
From: monkeyLoveding <125486029+monkeyLoveding@users.noreply.github.com>
Date: Mon, 20 Jul 2026 22:01:28 +0800
Subject: [PATCH] [NPU] memfabric-zbal update (#31777)

---
 docker/npu.Dockerfile                       | 4 +---
 scripts/ci/npu/npu_ci_install_dependency.sh | 3 +--
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/docker/npu.Dockerfile b/docker/npu.Dockerfile
index e82bdb1b5..e8922e23f 100644
--- a/docker/npu.Dockerfile
+++ b/docker/npu.Dockerfile
@@ -75,9 +75,7 @@ ENV LC_ALL=en_US.UTF-8
 RUN ${PIP_INSTALL} memfabric-hybrid==1.0.8
 
 ### Install zbal
-RUN if [ "$TARGETARCH" = "arm64" ]; then \
-      ${PIP_INSTALL} memfabric-zbal==1.1.1; \
-    fi
+RUN ${PIP_INSTALL} memfabric-zbal==1.2.0
 
 ### Install SGLang Model Gateway
 RUN ${PIP_INSTALL} sglang-router
diff --git a/scripts/ci/npu/npu_ci_install_dependency.sh b/scripts/ci/npu/npu_ci_install_dependency.sh
index 99ff530d8..8e2716d85 100755
--- a/scripts/ci/npu/npu_ci_install_dependency.sh
+++ b/scripts/ci/npu/npu_ci_install_dependency.sh
@@ -54,8 +54,7 @@ PTA_URL="https://gitcode.com/Ascend/pytorch/releases/download/v26.0.0-pytorch2.1
 ${PIP_INSTALL} ${PTA_URL}
 
 ### Install zbal
-${UV_PIP_INSTALL} memfabric-zbal==1.1.1
-
+${UV_PIP_INSTALL} memfabric-zbal==1.2.0
 ### Install Triton-Ascend
 ${PIP_INSTALL} triton-ascend==3.2.1.dev20260530 --extra-index-url=https://mirrors.huaweicloud.com/ascend/repos/pypi/nightly --trusted-host triton-ascend.osinfra.cn