[NPU] Update sgl-kernel-npu version to 2026.9.0 and move memfabric deps into pyproject (#37399)

This commit is contained in:
Even Zhou
2026-09-02 08:57:59 +08:00
committed by GitHub
parent 73a2414289
commit 9cc43ca4e1
5 changed files with 5 additions and 15 deletions
+1 -7
View File
@@ -41,10 +41,6 @@ export PATH="${CARGO_HOME:-$HOME/.cargo}/bin:${PATH}"
${UV_PIP_INSTALL} pybind11 pyyaml decorator scipy attrs psutil
### Install MemFabric
${UV_PIP_INSTALL} memfabric-hybrid==1.0.8
### Install PyTorch and PTA
PYTORCH_VERSION="2.10.0"
TORCHVISION_VERSION="0.25.0"
@@ -54,14 +50,12 @@ PTA_URL="https://gitcode.com/Ascend/pytorch/releases/download/v26.0.0-pytorch2.1
# GitCode does not allow UV downloads.
${PIP_INSTALL} ${PTA_URL}
### Install zbal
${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
### Install sgl-kernel-npu
SGLANG_KERNEL_NPU_TAG="20260826"
SGLANG_KERNEL_NPU_TAG="2026.9.0"
mkdir sgl-kernel-npu
(cd sgl-kernel-npu && wget "${GITHUB_PROXY_URL:=""}https://github.com/sgl-project/sgl-kernel-npu/releases/download/${SGLANG_KERNEL_NPU_TAG}/sgl-kernel-npu-${SGLANG_KERNEL_NPU_TAG}-torch${PYTORCH_VERSION}-py311-cann9.0.0-${DEVICE_TYPE}-$(arch).zip" \
&& unzip ./sgl-kernel-npu-${SGLANG_KERNEL_NPU_TAG}-torch${PYTORCH_VERSION}-py311-cann9.0.0-${DEVICE_TYPE}-$(arch).zip \