[NPU] chore: bump to CANN 8.3.RC1 and Pytorch 2.8.0 (#13647)

2025-11-21 17:07:08 +08:00
parent a244c0309c
commit 589d9ad55b
10 changed files with 66 additions and 83 deletions
@@ -311,9 +311,15 @@ class HIPEnv(BaseEnv):
 class NPUEnv(BaseEnv):
    """Environment checker for Ascend NPU"""

+    EXTRA_PACKAGE_LIST = [
+        "torch_npu",
+        "sgl-kernel-npu",
+        "deep_ep",
+    ]
+
    def __init__(self):
        super().__init__()
-        self.package_list = ["torch_npu", "sgl-kernel-npu"] + self.package_list
+        self.package_list.extend(NPUEnv.EXTRA_PACKAGE_LIST)

    def get_info(self):
        cuda_info = {"NPU available": torch.npu.is_available()}