Skip to content

Commit f3ad473

Browse files
danieldkdanieldk
authored
nix-builder: remove Torch 2.10 support (#570)
* nix-builder: remove Torch 2.10 support * ci: remove Torch 2.10 from tests * Update variant lists * ci: fix some more tests * ci: update tested ROCm version
1 parent 8c6364e commit f3ad473

18 files changed

Lines changed: 118 additions & 333 deletions

File tree

‎.github/workflows/build_kernel_macos.yaml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ jobs:
2828
# For now we only test that there are no regressions in building macOS
2929
# kernels. Also run tests once we have a macOS runner.
3030
- name: Build relu kernel
31-
run: ( cd examples/kernels/relu && nix build .\#redistributable.torch210-metal-aarch64-darwin -L )
31+
run: ( cd examples/kernels/relu && nix build .\#redistributable.torch211-metal-aarch64-darwin -L )
3232

3333
- name: Build relu metal cpp kernel
34-
run: ( cd examples/kernels/relu-metal-cpp && nix build .\#redistributable.torch210-metal-aarch64-darwin -L )
34+
run: ( cd examples/kernels/relu-metal-cpp && nix build .\#redistributable.torch211-metal-aarch64-darwin -L )

‎.github/workflows/build_kernel_rocm.yaml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ jobs:
3434
# For now we only test that there are no regressions in building ROCm
3535
# kernels. Also run tests once we have a ROCm runner.
3636
- name: Build relu kernel
37-
run: ( cd examples/kernels/relu && nix build .\#redistributable.torch210-cxx11-rocm70-x86_64-linux -L )
37+
run: ( cd examples/kernels/relu && nix build .\#redistributable.torch211-cxx11-rocm71-x86_64-linux -L )
3838

3939
- name: Build relu kernel (compiler flags)
40-
run: ( cd examples/kernels/relu-compiler-flags && nix build .\#redistributable.torch210-cxx11-rocm70-x86_64-linux )
40+
run: ( cd examples/kernels/relu-compiler-flags && nix build .\#redistributable.torch211-cxx11-rocm71-x86_64-linux )

‎.github/workflows/build_kernel_xpu.yaml‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,13 +34,13 @@ jobs:
3434
# For now we only test that there are no regressions in building XPU
3535
# kernels. Also run tests once we have a XPU runner.
3636
- name: Build relu kernel
37-
run: ( cd examples/kernels/relu && nix build .\#redistributable.torch210-cxx11-xpu20253-x86_64-linux -L )
37+
run: ( cd examples/kernels/relu && nix build .\#redistributable.torch211-cxx11-xpu20253-x86_64-linux -L )
3838

3939
- name: Build relu tvm-ffi kernel
4040
run: ( cd examples/kernels/relu-tvm-ffi && nix build .\#redistributable.tvm-ffi01-xpu20253-x86_64-linux -L )
4141

4242
- name: Build relu kernel (compiler flags)
43-
run: ( cd examples/kernels/relu-compiler-flags && nix build .\#redistributable.torch210-cxx11-xpu20253-x86_64-linux )
43+
run: ( cd examples/kernels/relu-compiler-flags && nix build .\#redistributable.torch211-cxx11-xpu20253-x86_64-linux )
4444

4545
- name: Build cutlass-gemm kernel
46-
run: ( cd examples/kernels/cutlass-gemm && nix build .\#redistributable.torch210-cxx11-xpu20253-x86_64-linux -L )
46+
run: ( cd examples/kernels/cutlass-gemm && nix build .\#redistributable.torch211-cxx11-xpu20253-x86_64-linux -L )

‎.github/workflows/test_kernels.yaml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@ jobs:
2626
strategy:
2727
max-parallel: 4
2828
matrix:
29-
python-version: ["3.10", "3.12"]
30-
torch-version: ["2.10.0", "2.11.0"]
29+
python-version: ["3.10", "3.14"]
30+
torch-version: ["2.11.0", "2.12.0"]
3131

3232
env:
3333
UV_PYTHON_PREFERENCE: only-managed

‎docs/source/builder/build-variants.md‎

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -7,27 +7,21 @@ available. This list will be updated as new PyTorch versions are released.
77

88
## CPU aarch64-darwin
99

10-
- `torch210-cpu-aarch64-darwin`
1110
- `torch211-cpu-aarch64-darwin`
1211
- `torch212-cpu-aarch64-darwin`
1312

1413
## Metal aarch64-darwin
1514

16-
- `torch210-metal-aarch64-darwin`
1715
- `torch211-metal-aarch64-darwin`
1816
- `torch212-metal-aarch64-darwin`
1917

2018
## CPU aarch64-linux
2119

22-
- `torch210-cxx11-cpu-aarch64-linux`
2320
- `torch211-cxx11-cpu-aarch64-linux`
2421
- `torch212-cxx11-cpu-aarch64-linux`
2522

2623
## CUDA aarch64-linux
2724

28-
- `torch210-cxx11-cu126-aarch64-linux`
29-
- `torch210-cxx11-cu128-aarch64-linux`
30-
- `torch210-cxx11-cu130-aarch64-linux`
3125
- `torch211-cxx11-cu126-aarch64-linux`
3226
- `torch211-cxx11-cu128-aarch64-linux`
3327
- `torch211-cxx11-cu130-aarch64-linux`
@@ -37,15 +31,11 @@ available. This list will be updated as new PyTorch versions are released.
3731

3832
## CPU x86_64-linux
3933

40-
- `torch210-cxx11-cpu-x86_64-linux`
4134
- `torch211-cxx11-cpu-x86_64-linux`
4235
- `torch212-cxx11-cpu-x86_64-linux`
4336

4437
## CUDA x86_64-linux
4538

46-
- `torch210-cxx11-cu126-x86_64-linux`
47-
- `torch210-cxx11-cu128-x86_64-linux`
48-
- `torch210-cxx11-cu130-x86_64-linux`
4939
- `torch211-cxx11-cu126-x86_64-linux`
5040
- `torch211-cxx11-cu128-x86_64-linux`
5141
- `torch211-cxx11-cu130-x86_64-linux`
@@ -55,16 +45,13 @@ available. This list will be updated as new PyTorch versions are released.
5545

5646
## ROCm x86_64-linux
5747

58-
- `torch210-cxx11-rocm70-x86_64-linux`
59-
- `torch210-cxx11-rocm71-x86_64-linux`
6048
- `torch211-cxx11-rocm71-x86_64-linux`
6149
- `torch211-cxx11-rocm72-x86_64-linux`
6250
- `torch212-cxx11-rocm71-x86_64-linux`
6351
- `torch212-cxx11-rocm72-x86_64-linux`
6452

6553
## XPU x86_64-linux
6654

67-
- `torch210-cxx11-xpu20253-x86_64-linux`
6855
- `torch211-cxx11-xpu20253-x86_64-linux`
6956
- `torch212-cxx11-xpu20253-x86_64-linux`
7057

‎examples/kernels/flake.nix‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@
8484
drv = sys: out: out.packages.${sys}.default;
8585
torchVersions = _defaultVersions: [
8686
{
87-
torchVersion = "2.10";
87+
torchVersion = "2.11";
8888
cudaVersion = "12.8";
8989
systems = [
9090
"x86_64-linux"

‎examples/kernels/relu-specific-torch/flake.nix‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
path = ./.;
1616
torchVersions = defaultVersions: [
1717
{
18-
torchVersion = "2.10";
18+
torchVersion = "2.11";
1919
cudaVersion = "12.8";
2020
systems = [
2121
"x86_64-linux"

‎examples/kernels/relu-torch-bounds/build.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@ backends = [
1111
repo-id = "kernels-test/relu-torch-bounds"
1212

1313
[torch]
14-
minver = "2.10"
15-
maxver = "2.10"
14+
minver = "2.11"
15+
maxver = "2.11"
1616
src = [
1717
"torch-ext/torch_binding.cpp",
1818
"torch-ext/torch_binding.h",

‎kernels/tests/kernel_locking/kernels.lock‎

Lines changed: 79 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -1,126 +1,178 @@
11
[
22
{
33
"repo_id": "kernels-community/relu",
4-
"sha": "468c458e7cfb386377d1f521bc2b8c77d1eb2562",
4+
"sha": "b0626dd282b2be0eaca55e216d3e39b519912f75",
55
"variants": {
66
"torch-ext": {
77
"hash": "sha256-fa95388531c6280130219f0e73e5daf521116da6c841fa5ab6a190c7994767d8",
88
"hash_type": "git_lfs_concat"
99
},
1010
"torch210-cpu-aarch64-darwin": {
11-
"hash": "sha256-e0492802726713ff5ec3362350d95ddde207c39c16a2ebad4bfc42a51b15b593",
11+
"hash": "sha256-28a2cea5757a8d310d0ea09f77e543791191ef4632bec5806460c3c85ddbfbab",
1212
"hash_type": "git_lfs_concat"
1313
},
1414
"torch210-cu128-x86_64-windows": {
1515
"hash": "sha256-a52a90e9caea7b8bcbb426cf94d92f9f9ab39b6aff41dfd69d4b9935e34709c0",
1616
"hash_type": "git_lfs_concat"
1717
},
1818
"torch210-cxx11-cpu-aarch64-linux": {
19-
"hash": "sha256-e2c191aa0bd6b761867f17c1264547b73f63bffda0f0e8131d3f253935e1c4a1",
19+
"hash": "sha256-13aaf95a93ca90956f4111271d30b9f4a6329a69e6fa008d8353d776c555e0c6",
2020
"hash_type": "git_lfs_concat"
2121
},
2222
"torch210-cxx11-cpu-x86_64-linux": {
23-
"hash": "sha256-12818400aa094e343ed02d94d61243fa5a81e222b1c3ef81d5a1b46026914e4c",
23+
"hash": "sha256-fbc59521ce8c0be4fd24cd011e24d6d999e8eca1871284af73500c9345ed3d13",
2424
"hash_type": "git_lfs_concat"
2525
},
2626
"torch210-cxx11-cu126-aarch64-linux": {
27-
"hash": "sha256-6b20165a41b9cd4fc72722d66dd9f12d9bdfee1eb5c3059cd115b11af906759f",
27+
"hash": "sha256-ad1f87c8d39bce651ecc949b5bf52107fbfceec5560a504e7cdb640dd167dd1a",
2828
"hash_type": "git_lfs_concat"
2929
},
3030
"torch210-cxx11-cu126-x86_64-linux": {
31-
"hash": "sha256-bc8ec6e982d9fdff7366090a6623ae1ad38f6f35eaacd792e23f10aa9e9ab7b6",
31+
"hash": "sha256-eeb332c30e4a65d2b7a43efe9c905480182b9f1332ce1154604a511220b7fde8",
3232
"hash_type": "git_lfs_concat"
3333
},
3434
"torch210-cxx11-cu128-aarch64-linux": {
35-
"hash": "sha256-8cd1a1e36477a224b4de6441440fa3265672a7cfbb9289f57b11800ef3e809bd",
35+
"hash": "sha256-ba31e1501ba03c6f9457cc22de7e1a33b4b0f7563a31bdefe78d59e9f1f1a3c3",
3636
"hash_type": "git_lfs_concat"
3737
},
3838
"torch210-cxx11-cu128-x86_64-linux": {
39-
"hash": "sha256-c0be23298e650e705b5b7cb6db00fb65b8beeee39dcbc0718221f16802eccc07",
39+
"hash": "sha256-a1917971ab09d73c28089a8fa9f3c6a6634a6817fecd6fbc6167bb6983451642",
4040
"hash_type": "git_lfs_concat"
4141
},
4242
"torch210-cxx11-cu130-aarch64-linux": {
43-
"hash": "sha256-a1bc582a87834c6728c50f69f8a954d03e3dc73cce2f775d0bc6f1c42a1d40dd",
43+
"hash": "sha256-6f5dd29d726471608e4e3b7f56926545582ad875ed143fa1789dd0a76f26787a",
4444
"hash_type": "git_lfs_concat"
4545
},
4646
"torch210-cxx11-cu130-x86_64-linux": {
47-
"hash": "sha256-cfbcb3fe7c3417c64502f84b97ddeb4d4f6c2e9bd1bc7d155accbf19682e5467",
47+
"hash": "sha256-f22fdd300b80ba08a65546e7a12847d22faafcb92839e09fb67612e3018f24ec",
4848
"hash_type": "git_lfs_concat"
4949
},
5050
"torch210-cxx11-rocm70-x86_64-linux": {
51-
"hash": "sha256-ef3061f180a935be046caaedbe8eb3c9611e5a4228d88d1100f6357a3b5aa46c",
51+
"hash": "sha256-89e26a1d8215bfc11228905c26d20d3519e0d09ba5306496f4069596e66fb7a6",
5252
"hash_type": "git_lfs_concat"
5353
},
5454
"torch210-cxx11-rocm71-x86_64-linux": {
55-
"hash": "sha256-4ec58bc0dd36b98f1153ad2c870906995a5f13278273a5ce9eda861047380ec7",
55+
"hash": "sha256-a07fc6bb7632a79f5eab1734d45070b7e0cc93deae928a1c410c2ed4a0a2c62c",
5656
"hash_type": "git_lfs_concat"
5757
},
5858
"torch210-cxx11-xpu20253-x86_64-linux": {
59-
"hash": "sha256-aa4eaa81b2f260f6e35d8e7ab4f1aca5cdb91bcb49ed8c8f641820919c8209ab",
59+
"hash": "sha256-67cdfa652f832b5d630824036ebc584cda438ad12030241e83cf46bad64143d9",
6060
"hash_type": "git_lfs_concat"
6161
},
6262
"torch210-metal-aarch64-darwin": {
63-
"hash": "sha256-424c70191e4808aec776987b3b5ce9ce184348513ce703e200593a71b40cbea3",
63+
"hash": "sha256-e89ea77566a2744e83e545dc120b3e580295eaf34bf053354924296cb49bdb07",
6464
"hash_type": "git_lfs_concat"
6565
},
6666
"torch210-xpu20253-x86_64-windows": {
6767
"hash": "sha256-47f3e41d16251a45edaf5d80c8de531cf037bf7d64cb25c6f9632a33d33b669e",
6868
"hash_type": "git_lfs_concat"
6969
},
7070
"torch211-cpu-aarch64-darwin": {
71-
"hash": "sha256-c6cda4b2f0f83f4a86939575e9dffcd74d6cdaa6b89da540e86e8d536156245b",
71+
"hash": "sha256-27b2935b5378830f8fb4b03a8d808aad584182f240f09907f08f8c54389b0feb",
7272
"hash_type": "git_lfs_concat"
7373
},
7474
"torch211-cu128-x86_64-windows": {
7575
"hash": "sha256-f2367fb89a5995458ed0c5011c7ba69b9b0f2fc43eb434b5b217a7b5ee890783",
7676
"hash_type": "git_lfs_concat"
7777
},
7878
"torch211-cxx11-cpu-aarch64-linux": {
79-
"hash": "sha256-2ec67a9846bba2ac8a053b4f11d00e086315d4e028b94e5e8168ed00444ee5da",
79+
"hash": "sha256-d9ea423b06708d8c318b68daa9e570c75309904bdf88f2c6921a8a3351ea239e",
8080
"hash_type": "git_lfs_concat"
8181
},
8282
"torch211-cxx11-cpu-x86_64-linux": {
83-
"hash": "sha256-4f9c7dcb0e97ccb0507dd80881c7fc5446fb3626856a42f172c40c65557b064f",
83+
"hash": "sha256-0e45e43d0fd4d81e09d190857e05934e0d24895f3110ee8d78ebf8f9eb37031a",
8484
"hash_type": "git_lfs_concat"
8585
},
8686
"torch211-cxx11-cu126-aarch64-linux": {
87-
"hash": "sha256-7b9ec9ae8d9ceb5af1c9be1517cd0de989b1f1516bb6ff956e435224302f3cf6",
87+
"hash": "sha256-a226ff64d76eed9b7fded263f2386b11d4ea7daffc4853b47d64ff8acd67e49e",
8888
"hash_type": "git_lfs_concat"
8989
},
9090
"torch211-cxx11-cu126-x86_64-linux": {
91-
"hash": "sha256-de3d3cbdb9817773637da80fab88870e7ad9ce7d7e40899b3a4e5cd7182dc645",
91+
"hash": "sha256-1d4c1d36926c3bb7ee549421d8fff4bd2ad23ec6d7f7c8a4717523e2f50b4655",
9292
"hash_type": "git_lfs_concat"
9393
},
9494
"torch211-cxx11-cu128-aarch64-linux": {
95-
"hash": "sha256-559aeba2d57dd3125984c8dd1b1e5358b6807d9b8f1fdc165519e6f1c5e33984",
95+
"hash": "sha256-0b8314b90266b6bee67d4b8bdfb1c2c54bc7b1c8186372ae9e4cd7f8bcc09f04",
9696
"hash_type": "git_lfs_concat"
9797
},
9898
"torch211-cxx11-cu128-x86_64-linux": {
99-
"hash": "sha256-f078aa837db738094599b395372dfb50982dfbeb2538b36f2f2d2b3eec127a51",
99+
"hash": "sha256-dc699374e204035efe1e3e53b6440c2ce0a0e799a5486cc1d680611e9a761cc7",
100100
"hash_type": "git_lfs_concat"
101101
},
102102
"torch211-cxx11-cu130-aarch64-linux": {
103-
"hash": "sha256-7ed13391cef18c36cc446d210d2fd8592993d45228a674182ee2cff375fa05ae",
103+
"hash": "sha256-6d242c9431049d913a3a5a12ad642cf1e681230cba05d777d5ff606f0414c5d0",
104104
"hash_type": "git_lfs_concat"
105105
},
106106
"torch211-cxx11-cu130-x86_64-linux": {
107-
"hash": "sha256-42019ed67eff74f5168a2e9d6bf276d047b29c11d5e3395a342893a3f50daae1",
107+
"hash": "sha256-b65cf05ea577f1806a1da98bb1103f42e88a85b309984fc8cce8bdb7c663d9ed",
108108
"hash_type": "git_lfs_concat"
109109
},
110110
"torch211-cxx11-rocm71-x86_64-linux": {
111-
"hash": "sha256-d82aff83823ac9ee81c78db67b860b40d9cbbd6b7efe8640693afdbe16f3e52c",
111+
"hash": "sha256-5e56897157f3be71df46a8f1f4e1797e2dde1a8c0b28d04fb074fba87ef03eae",
112112
"hash_type": "git_lfs_concat"
113113
},
114114
"torch211-cxx11-rocm72-x86_64-linux": {
115-
"hash": "sha256-772ec7f9b0917421f562d162b7db8535cac09b92cc2ce670ba467a08f1dfb9b0",
115+
"hash": "sha256-35f1ba226d649250752973dccb4ab0c159be3ee7f64476849c4f8f8bdd598414",
116116
"hash_type": "git_lfs_concat"
117117
},
118118
"torch211-cxx11-xpu20253-x86_64-linux": {
119-
"hash": "sha256-4466ae2451d960a6c57a456d0c6a5f7397ccabcc6fb43fc3c0dcf26aecda5cfa",
119+
"hash": "sha256-15852983def8d8da3ee691233bef26168e3832063623923b3991ef3a3a2dad68",
120120
"hash_type": "git_lfs_concat"
121121
},
122122
"torch211-metal-aarch64-darwin": {
123-
"hash": "sha256-b9bb76b444aae35782626327647f77f651f67ec666b58f1d3a8f2f251af88a50",
123+
"hash": "sha256-7978552e3c54d5ec41dfe1f27a990b40cb6ce4672055266b588029a3227f5a13",
124+
"hash_type": "git_lfs_concat"
125+
},
126+
"torch212-cpu-aarch64-darwin": {
127+
"hash": "sha256-f8c35fc430e68525fb204a632c633b206993fe5140d805afb25599567c900c06",
128+
"hash_type": "git_lfs_concat"
129+
},
130+
"torch212-cxx11-cpu-aarch64-linux": {
131+
"hash": "sha256-417fb435907db70523f82e84872fd0fdf736157de7e26919734effd32702637d",
132+
"hash_type": "git_lfs_concat"
133+
},
134+
"torch212-cxx11-cpu-x86_64-linux": {
135+
"hash": "sha256-68bb6d95f644b64f8d725dca14c60d4633edbce181e157195f343d8a8df97148",
136+
"hash_type": "git_lfs_concat"
137+
},
138+
"torch212-cxx11-cu126-aarch64-linux": {
139+
"hash": "sha256-12344fd63b625f4a4279c95f1c312c9f72bc04843022c094fc748861ca63a006",
140+
"hash_type": "git_lfs_concat"
141+
},
142+
"torch212-cxx11-cu126-x86_64-linux": {
143+
"hash": "sha256-c521aed8ce128b985cdb476d4ad55d0583ccb5acaa3a01d31a8396d7c2fc78f2",
144+
"hash_type": "git_lfs_concat"
145+
},
146+
"torch212-cxx11-cu130-aarch64-linux": {
147+
"hash": "sha256-1676a87551048c4488328f128bbe4b9e1fe29b07cc2fed9e94319445cecb85ad",
148+
"hash_type": "git_lfs_concat"
149+
},
150+
"torch212-cxx11-cu130-x86_64-linux": {
151+
"hash": "sha256-2d1354a462ebbfb9eeebfa350f03c3668572b5cf18b9864e458316108c9c5d1e",
152+
"hash_type": "git_lfs_concat"
153+
},
154+
"torch212-cxx11-cu132-aarch64-linux": {
155+
"hash": "sha256-c9f2953832d56a7ec4afdf1551187e4aeaf7f8acbd6c1fbaa0c5b766e6b5fa82",
156+
"hash_type": "git_lfs_concat"
157+
},
158+
"torch212-cxx11-cu132-x86_64-linux": {
159+
"hash": "sha256-669e6894ae35c3e004b841fd2e35cd579979178a4fb547eec97a190062a68488",
160+
"hash_type": "git_lfs_concat"
161+
},
162+
"torch212-cxx11-rocm71-x86_64-linux": {
163+
"hash": "sha256-70db275da9110da0d2cba7579006fdf4a919ae08f4254c22326757d5db928e21",
164+
"hash_type": "git_lfs_concat"
165+
},
166+
"torch212-cxx11-rocm72-x86_64-linux": {
167+
"hash": "sha256-dfe26a58f8560f11c4e82acb56094b43fbb72ef032a3b61922fa3188a27d9627",
168+
"hash_type": "git_lfs_concat"
169+
},
170+
"torch212-cxx11-xpu20253-x86_64-linux": {
171+
"hash": "sha256-3237e8218022e8aad1cf995965af7a678e62374e47b1dadff935eb54e53ea0ca",
172+
"hash_type": "git_lfs_concat"
173+
},
174+
"torch212-metal-aarch64-darwin": {
175+
"hash": "sha256-44a87b3557317554fc06b377174d7c37165a7feadf07417ec9bd2cb2ed29f96b",
124176
"hash_type": "git_lfs_concat"
125177
},
126178
"torch27-cxx11-cu118-x86_64-linux": {

0 commit comments

Comments
 (0)