pioduino/uv
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Sort packages in GPU lists (#17054)

Browse Source 
## 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
1 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/uv-torch/src/backend.rs
+12 -12
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"
)
}