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