@@ -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
103103fsdp = [
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