Python
/
uv
mirror of https://github.com/astral-sh/uv
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Sort packages in GPU lists (#17054) 

## Summary

No semantic changes; this is just bothering me.
This commit is contained in:
Charlie Marsh 2025-12-09 15:10:06 -05:00 committed by GitHub
parent d0a6f5d13f
commit 2502577c9d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
crates/uv-torch/src/backend.rs
View File

@ -310,7 +310,10 @@ impl TorchStrategy {
TorchSource::PyTorch => {
matches!(
package_name.as_str(),
"torch"
"pytorch-triton"
| "pytorch-triton-rocm"
| "pytorch-triton-xpu"
| "torch"
| "torcharrow"
| "torchaudio"
| "torchcsprng"
@ -320,9 +323,6 @@ impl TorchStrategy {
| "torchtext"
| "torchvision"
| "triton"
| "pytorch-triton"
| "pytorch-triton-rocm"
| "pytorch-triton-xpu"
)
}
TorchSource::Pyx => {
@ -334,12 +334,14 @@ impl TorchStrategy {
| "megablocks"
| "natten"
| "pyg-lib"
| "pytorch-triton"
| "pytorch-triton-rocm"
| "pytorch-triton-xpu"
| "torch"
| "torch-cluster"
| "torch-scatter"
| "torch-sparse"
| "torch-spline-conv"
| "vllm"
| "torch"
| "torcharrow"
| "torchaudio"
| "torchcsprng"
@ -349,9 +351,7 @@ impl TorchStrategy {
| "torchtext"
| "torchvision"
| "triton"
| "pytorch-triton"
| "pytorch-triton-rocm"
| "pytorch-triton-xpu"
| "vllm"
)
}
}
@ -365,12 +365,11 @@ impl TorchStrategy {
pub fn has_system_dependency(&self, package_name: &PackageName) -> bool {
matches!(
package_name.as_str(),
"flash-attn"
"deepspeed"
| "flash-attn"
| "flash-attn-3"
| "megablocks"
| "natten"
| "deepspeed"
| "vllm"
| "torch"
| "torcharrow"
| "torchaudio"
@ -379,6 +378,7 @@ impl TorchStrategy {
| "torchdistx"
| "torchtext"
| "torchvision"
| "vllm"
)
}