Skip to content

Commit 48f5076

Browse files
Add PyG packages to torch backend (#15911)
## Summary These are now supported on pyx.
1 parent d4806ee commit 48f5076

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

crates/uv-torch/src/backend.rs

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -322,11 +322,16 @@ impl TorchStrategy {
322322
TorchSource::Pyx => {
323323
matches!(
324324
package_name.as_str(),
325-
"flash-attn"
325+
"deepspeed"
326+
| "flash-attn"
326327
| "flash-attn-3"
327328
| "megablocks"
328329
| "natten"
329-
| "deepspeed"
330+
| "pyg-lib"
331+
| "torch-cluster"
332+
| "torch-scatter"
333+
| "torch-sparse"
334+
| "torch-spline-conv"
330335
| "vllm"
331336
| "torch"
332337
| "torch-model-archiver"

0 commit comments

Comments
 (0)