Update to final Torch 2.13.0, cleanup - #732
Conversation
|
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
In preparation of TPU support, which is still at Torch 2.11.
Coverage report —
|
| Name | Stmts | Miss | Cover | Missing |
|---|---|---|---|---|
| src/kernels/__init__.py | 10 | 0 | 100% | |
| src/kernels/_system.py | 6 | 1 | 83% | 10 |
| src/kernels/_versions.py | 63 | 7 | 89% | 46, 49, 52-53, 56-57, 100 |
| src/kernels/backends.py | 194 | 55 | 72% | 40, 44, 48-51, 68, 90, 108, 117, 121, 125-127, 148, 170, 181, 188-191, 201, 205-225, 233, 256-276 |
| src/kernels/compat.py | 8 | 1 | 88% | 5 |
| src/kernels/deps.py | 54 | 4 | 93% | 58-59, 95, 98 |
| src/kernels/layer/__init__.py | 6 | 0 | 100% | |
| src/kernels/layer/_interval_tree.py | 103 | 4 | 96% | 23, 52, 147, 150 |
| src/kernels/layer/device.py | 48 | 14 | 71% | 42, 47-49, 91, 96-98, 101, 149, 152, 155-157 |
| src/kernels/layer/func.py | 81 | 7 | 91% | 81, 111, 183, 301, 307, 320, 338 |
| src/kernels/layer/globals.py | 5 | 0 | 100% | |
| src/kernels/layer/kernelize.py | 74 | 8 | 89% | 255, 281, 289-290, 296, 300, 316-318 |
| src/kernels/layer/layer.py | 210 | 16 | 92% | 167, 210, 216, 229, 337, 417-418, 430, 439, 447, 458, 487, 491, 504, 557, 587 |
| src/kernels/layer/mode.py | 14 | 0 | 100% | |
| src/kernels/layer/repos.py | 130 | 34 | 74% | 27, 33, 36-41, 61-62, 68, 71-74, 88, 92, 101-102, 108, 111-114, 121-122, 128, 131-134, 141-142, 148, 151-154, 235 |
| src/kernels/lockfile.py | 71 | 46 | 35% | 37-104, 108-131 |
| src/kernels/status.py | 49 | 2 | 96% | 23, 81 |
| src/kernels/utils.py | 313 | 55 | 82% | 65, 77-81, 87-88, 246, 250, 253, 315, 323, 362-363, 401, 432, 437, 472, 701, 704, 706, 712, 725-726, 747-759, 763-770, 778, 782-792, 796-803, 841, 845, 864, 866 |
| src/kernels/variants.py | 270 | 22 | 92% | 63, 94, 115, 145, 254-255, 297-300, 302, 386-393, 399-405, 436-442, 454-460, 559-561 |
| src/kernels/verify.py | 88 | 1 | 99% | 32 |
| TOTAL | 1797 | 277 | 85% |
Updated by the Test kernels workflow on commit eba33e2766798f08936ae897fdbd6bc1d622a8ff.
| # Remove once sglang moves to a newer Torch version. | ||
| torch-bin_2_9 = mkTorch { | ||
| version = "2.9"; |
There was a problem hiding this comment.
Have they moved? I guess we needed it for building purposes?
There was a problem hiding this comment.
Opened an internal discussion thread - sglang is on Torch 2.11 now (which we are also deprecating now). However, the kernel already has all the bits to build against the Torch stable API. It's mostly a matter of making some changes in build.toml/flake.nix and ensuring that the API in flash_api_stable.cpp is in sync with flash_api.cpp modulo stable API/ABI differences.
| url = "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.13.0/rc15/triton_xpu-3.7.2-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"; | ||
| url = "https://download.pytorch.org/whl/triton_xpu-3.7.2-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"; |
There was a problem hiding this comment.
I didn't notice this before. Jazzy.
| } | ||
| } | ||
| }, | ||
| "2.11": { |
Uh oh!
There was an error while loading. Please reload this page.