Skip to content

Commit b3cd1ac

Browse files
authored
[chore] pin nixl-cu12 (NovaSky-AI#1759)
NovaSky-AI#1756 was insufficient, since the marker on override-dependencies gates when the override happens not when the dependency is installed.
1 parent 689d119 commit b3cd1ac

2 files changed

Lines changed: 49 additions & 88 deletions

File tree

pyproject.toml

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -97,14 +97,14 @@ skyrl-train = [
9797
"vllm-router; sys_platform == 'linux'",
9898
"pybind11",
9999
"setuptools",
100-
"nixl; sys_platform == 'linux'",
100+
"nixl-cu12; sys_platform == 'linux'",
101101
]
102102

103103
fsdp = [
104104
"skyrl[skyrl-train]",
105105
"vllm==0.20.2; sys_platform == 'linux'",
106106
"vllm-router; sys_platform == 'linux'",
107-
"nixl; sys_platform == 'linux'",
107+
"nixl-cu12; sys_platform == 'linux'",
108108
"flash-linear-attention; sys_platform == 'linux'",
109109
"causal-conv1d; sys_platform == 'linux'",
110110
"flash-attn==2.8.3; sys_platform == 'linux'",
@@ -124,7 +124,7 @@ megatron = [
124124
"mamba-ssm>=2.3.0; sys_platform == 'linux'",
125125
"vllm==0.20.2; sys_platform == 'linux'",
126126
"vllm-router; sys_platform == 'linux'",
127-
"nixl; sys_platform == 'linux'",
127+
"nixl-cu12; sys_platform == 'linux'",
128128
"torch==2.11.0; sys_platform == 'linux'",
129129
"flashinfer-python==0.6.8.post1; sys_platform == 'linux' and platform_machine == 'x86_64'",
130130
"torchvision; sys_platform == 'linux'",
@@ -224,11 +224,6 @@ override-dependencies = [
224224
"transformers>=5.6.1,<=5.8.0; sys_platform == 'linux'",
225225
"megatron-core>=0.16.0; sys_platform == 'linux'",
226226
"ml_dtypes>=0.5.0; sys_platform == 'linux'",
227-
# The `nixl` meta-package pulls BOTH nixl-cu12 and nixl-cu13 on Linux; both
228-
# ship a top-level `nixl_ep` that collide, and cu13 (libcudart.so.13) wins,
229-
# breaking vLLM model inspection on this CUDA-12 env (torch cu128 / vllm
230-
# cu129). Exclude cu13 so only nixl-cu12 (libcudart.so.12) is installed.
231-
"nixl-cu13; sys_platform == 'never'",
232227
]
233228

234229
[tool.uv.extra-build-dependencies]

0 commit comments

Comments
 (0)