Commit 523f46b
committed
fix: use --torch-backend instead of --extra-index-url for dependency compilation
The old --extra-index-url approach caused uv to resolve any package from
the PyTorch wheel index when it happened to be available there, not just
torch-ecosystem packages.
Replace it with uv's --torch-backend flag which routes only torch,
torchaudio, torchvision, and torchsde to the correct PyTorch index.
Map NVIDIA to cu126, AMD to rocm6.1, and CPU to the cpu index so each
backend gets the right torch variant.
Pin uv>=0.6.0 in pyproject.toml as the minimum version supporting
--torch-backend.1 parent 8bab84d commit 523f46b
2 files changed
Lines changed: 6 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
64 | 64 | | |
65 | 65 | | |
66 | 66 | | |
| 67 | + | |
67 | 68 | | |
68 | 69 | | |
69 | 70 | | |
| 71 | + | |
70 | 72 | | |
71 | 73 | | |
72 | 74 | | |
| |||
387 | 389 | | |
388 | 390 | | |
389 | 391 | | |
| 392 | + | |
390 | 393 | | |
391 | 394 | | |
392 | 395 | | |
393 | 396 | | |
394 | 397 | | |
| 398 | + | |
395 | 399 | | |
396 | 400 | | |
397 | 401 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
94 | 94 | | |
95 | 95 | | |
96 | 96 | | |
97 | | - | |
98 | | - | |
| 97 | + | |
| 98 | + | |
99 | 99 | | |
100 | 100 | | |
101 | 101 | | |
| |||
0 commit comments