[NVIDIA] update pyproject.toml to support cu130 option (#11521)
This commit is contained in:
2
.github/workflows/release-docker-dev.yml
vendored
2
.github/workflows/release-docker-dev.yml
vendored
@@ -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:
|
||||||
|
|||||||
2
.github/workflows/release-docker.yml
vendored
2
.github/workflows/release-docker.yml
vendored
@@ -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 = [
|
|||||||
"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 = [
|
|||||||
]
|
]
|
||||||
|
|
||||||
[project.optional-dependencies]
|
[project.optional-dependencies]
|
||||||
decord = ["decord"]
|
decord = ["decord2"]
|
||||||
test = [
|
test = [
|
||||||
"accelerate",
|
"accelerate",
|
||||||
"expecttest",
|
"expecttest",
|
||||||
@@ -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]"]
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user