Skip to content

Commit 1321d0e

Browse files
committed
Publish cuda-pathfinder to the torch CUDA arch indexes
torch declares cuda-bindings, and cuda-bindings requires cuda-pathfinder>=1.4.2. Release smoke tests install from a single arch index with no PyPI fallback, so that transitive dependency has to be resolvable there too. cuda-pathfinder was registered under the vllm project only, so update_dependencies.py never wrote whl/{test,nightly}/cu*/cuda-pathfinder/. whl/test/cu134 advertises it in the PEP 503 listing while the target 403s. The older arches still serve it, but that is residue from before it stopped being produced -- nothing regenerates it today, so cu134 is simply the first arch created after that. Give it the same torch targets as cuda-bindings, since it exists to satisfy cuda-bindings. cuda-python is left alone: torch never depends on it, so its absence from the arch indexes is correct.
1 parent c22f7fa commit 1321d0e

1 file changed

Lines changed: 27 additions & 1 deletion

File tree

s3_management/update_dependencies.py

Lines changed: 27 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -768,7 +768,33 @@
768768
# vLLM
769769
"ninja": [{"project": "vllm"}],
770770
"cuda-python": [{"project": "vllm"}],
771-
"cuda-pathfinder": [{"project": "vllm"}],
771+
"cuda-pathfinder": [
772+
{
773+
"project": "torch",
774+
"target": "cu126",
775+
},
776+
{
777+
"project": "torch",
778+
"target": "cu128",
779+
},
780+
{
781+
"project": "torch",
782+
"target": "cu129",
783+
},
784+
{
785+
"project": "torch",
786+
"target": "cu130",
787+
},
788+
{
789+
"project": "torch",
790+
"target": "cu132",
791+
},
792+
{
793+
"project": "torch",
794+
"target": "cu134",
795+
},
796+
{"project": "vllm"},
797+
],
772798
"pynvml": [{"project": "vllm"}],
773799
"nvidia-ml-py": [{"project": "vllm"}],
774800
"einops": [{"project": "vllm"}],

0 commit comments

Comments
 (0)