Unverified Commit cb8f3d90 authored by Johnny's avatar Johnny Committed by GitHub
Browse files

[NVIDIA] update pyproject.toml to support cu130 option (#11521)

parent 4b694e7d
...@@ -19,7 +19,7 @@ jobs: ...@@ -19,7 +19,7 @@ jobs:
version: 12.9.1 version: 12.9.1
- runner: arm-docker-build-node - runner: arm-docker-build-node
platform: linux/arm64 platform: linux/arm64
build_type: all_aarch64 build_type: all
tag: dev-arm64 tag: dev-arm64
version: 12.9.1 version: 12.9.1
steps: steps:
......
...@@ -66,7 +66,7 @@ jobs: ...@@ -66,7 +66,7 @@ jobs:
matrix: matrix:
variant: variant:
- cuda_version: "12.9.1" - cuda_version: "12.9.1"
build_type: "all_aarch64" build_type: "all"
runs-on: arm-docker-build-node runs-on: arm-docker-build-node
steps: steps:
- name: Delete huge unnecessary tools folder - name: Delete huge unnecessary tools folder
......
...@@ -46,7 +46,7 @@ dependencies = [ ...@@ -46,7 +46,7 @@ dependencies = [
"py-spy", "py-spy",
"pybase64", "pybase64",
"pydantic", "pydantic",
"pynvml", "nvidia-ml-py",
"python-multipart", "python-multipart",
"pyzmq>=25.1.2", "pyzmq>=25.1.2",
"requests", "requests",
...@@ -73,7 +73,7 @@ dependencies = [ ...@@ -73,7 +73,7 @@ dependencies = [
] ]
[project.optional-dependencies] [project.optional-dependencies]
decord = ["decord"] decord = ["decord2"]
test = [ test = [
"accelerate", "accelerate",
"expecttest", "expecttest",
...@@ -93,11 +93,21 @@ tracing = [ ...@@ -93,11 +93,21 @@ tracing = [
"opentelemetry-sdk", "opentelemetry-sdk",
] ]
all = ["sglang[test]", "sglang[decord]"] all = ["sglang[test]", "sglang[decord]"]
all_aarch64 = ["sglang[test]"] cu130 = [
dev = ["sglang[test]", "sglang[decord]"] "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 # The following will be deprecated in 2 weeks
dev = ["sglang[test]", "sglang[decord]"]
all_aarch64 = ["sglang[test]"]
blackwell = ["sglang[test]", "sglang[decord]"] blackwell = ["sglang[test]", "sglang[decord]"]
blackwell_aarch64 = ["sglang[test]"] blackwell_aarch64 = ["sglang[test]"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment