[NPU] Update sgl-kernel-npu version to 2026.9.0 and move memfabric deps into pyproject (#37399)
This commit is contained in:
@@ -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 \
|
||||
|
||||
Reference in New Issue
Block a user