From cb8f3d90d33f8d3c54d459f9e661922c75212e9c Mon Sep 17 00:00:00 2001
From: Johnny <johnnynuca14@gmail.com>
Date: Mon, 13 Oct 2025 22:03:31 +0200
Subject: [PATCH] [NVIDIA] update pyproject.toml to support cu130 option
 (#11521)

---
 .github/workflows/release-docker-dev.yml |  2 +-
 .github/workflows/release-docker.yml     |  2 +-
 python/pyproject.toml                    | 18 ++++++++++++++----
 3 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/release-docker-dev.yml b/.github/workflows/release-docker-dev.yml
index 4564b981f..f275ed834 100644
--- a/.github/workflows/release-docker-dev.yml
+++ b/.github/workflows/release-docker-dev.yml
@@ -19,7 +19,7 @@ jobs:
             version: 12.9.1
           - runner: arm-docker-build-node
             platform: linux/arm64
-            build_type: all_aarch64
+            build_type: all
             tag: dev-arm64
             version: 12.9.1
     steps:
diff --git a/.github/workflows/release-docker.yml b/.github/workflows/release-docker.yml
index 4c12bc81c..30e64b014 100644
--- a/.github/workflows/release-docker.yml
+++ b/.github/workflows/release-docker.yml
@@ -66,7 +66,7 @@ jobs:
       matrix:
         variant:
           - cuda_version: "12.9.1"
-            build_type: "all_aarch64"
+            build_type: "all"
     runs-on: arm-docker-build-node
     steps:
       - name: Delete huge unnecessary tools folder
diff --git a/python/pyproject.toml b/python/pyproject.toml
index 4061f8576..944a81ec5 100755
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -46,7 +46,7 @@ dependencies = [
   "py-spy",
   "pybase64",
   "pydantic",
-  "pynvml",
+  "nvidia-ml-py",
   "python-multipart",
   "pyzmq>=25.1.2",
   "requests",
@@ -73,7 +73,7 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-decord = ["decord"]
+decord = ["decord2"]
 test = [
   "accelerate",
   "expecttest",
@@ -93,11 +93,21 @@ tracing = [
   "opentelemetry-sdk",
 ]
 all = ["sglang[test]", "sglang[decord]"]
-all_aarch64 = ["sglang[test]"]
-dev = ["sglang[test]", "sglang[decord]"]
+cu130 = [
+  "torch==2.9.0",
+  "torchaudio==2.9.0",
+  "torchvision==0.24.0",
+]
+cu130_all = [
+  "sglang[test]",
+  "sglang[decord]",
+  "sglang[cu130]"
+]
 
 
 # The following will be deprecated in 2 weeks
+dev = ["sglang[test]", "sglang[decord]"]
+all_aarch64 = ["sglang[test]"]
 blackwell = ["sglang[test]", "sglang[decord]"]
 blackwell_aarch64 = ["sglang[test]"]