Skip to content

[release/2.11] Fix ROCm Triton wheel version to match torch's triton pin #3477

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
ethanwee1 wants to merge 5 commits into
base: release/2.11
Choose a base branch
from
Draft

[release/2.11] Fix ROCm Triton wheel version to match torch's triton pin #3477

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
6c13834
[release/2.11] Fix ROCm Triton wheel version to match torch's triton pin
ethanwee1 Jul 23, 2026
3e2707b
[release/2.11] Test triton-side version fix: bump triton pin, drop bu…
ethanwee1 Jul 23, 2026
c199b83
[release/2.11] Bump AOTriton pin 0.12.50tp2 -> 0.13b to fix GCC build…
ethanwee1 Jul 23, 2026
5bf3cd0
[TEST ONLY] Point torchaudio related_commits at fork with sources-str…
ethanwee1 Jul 23, 2026
c946bbe
[TEST ONLY] Point torchaudio related_commits at ROCm/audio sources-st…
ethanwee1 Jul 23, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .ci/docker/ci_commit_pins/triton.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1 +1 @@
4cff872ced001ea92d9fcf05b3f6517e2b486d19
72c9556a181aa9b6775b1ca081809fb23d1d7527
31 changes: 17 additions & 14 deletions cmake/External/aotriton.cmake
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,32 +9,35 @@ if(NOT __AOTRITON_INCLUDED)
# Replaces .ci/docker/aotriton_version.txt
# Note packages information may have versions skipped (due to no ABI breaks)
# But they must be listed from lower version to higher version
set(__AOTRITON_VER "0.12.50tp2")
set(__AOTRITON_VER "0.13b")
set(__AOTRITON_MANYLINUX_LIST
"manylinux_2_28" # rocm6.4
"manylinux_2_28" # rocm7.0
"manylinux_2_28" # rocm7.1
"manylinux_2_28" # rocm7.2
"manylinux_2_28" # rocm7.14
"manylinux_2_28" # rocm7.15
)
set(__AOTRITON_ROCM_LIST
"rocm6.4"
"rocm7.0"
"rocm7.1"
"rocm7.2"
"rocm7.14"
"rocm7.15"
)
if(DEFINED ENV{PYTORCH_AOTRITON_COMMIT})
set(__AOTRITON_CI_COMMIT "$ENV{PYTORCH_AOTRITON_COMMIT}")
else()
set(__AOTRITON_CI_COMMIT "05f26c2d8fbdcf8dd6d95b9544d04f7a39fc9920")
set(__AOTRITON_CI_COMMIT "6e00ef3e335b45dfb49065259533b59c68995bfe")
endif()
set(__AOTRITON_SHA256_LIST
"70020f5938d84e2dbb7cd98a4ca9cb46c4c0a61b683611c61dac20e99c1ee377" # rocm6.4
"2a334e29f316e5a40766dbcf0b3075f7c73cce57f5044b39b7c3a7b33a7826bd" # rocm7.0
"cc5894fed1dd44464a6187cafb82e61d9e4034b3035257c0fdfa93d4f0e906cc" # rocm7.1
"af4ca6c5889c7ba302659115051ea1bfe6a09fb89f311dae214a6f942e93f605" # rocm7.2
"21895066db0c0e2079f5483d07fbfde659f8de906c589bbfbc774bb9b4565e11" # rocm7.14
"2fafa80953d9a49bd20e794bb8c0e1646e8aa815be2fb161deaa849a47547b17" # rocm6.4
"7409f7c974cc79be731a419818bdb2ed6b8a3640fd40665baa76ec3c2a537204" # rocm7.0
"f061a997679d8529a7b196b0ffb39912145ede217e515e1ee9ef5673b56d9e41" # rocm7.1
"1cdeebb7ef61ab691fba1d81da919b9db5d8bef28269c892a30bd13a0495b7a0" # rocm7.2
"7a139797c16b002fd5d9bcd706d36dc9819bb108877150f8186da21d0590eaa6" # rocm7.14
"f024225d8b6063f7d95974e5957cb20893a1579a9a73b22b60426441331bc021" # rocm7.15
)
set(__AOTRITON_IMAGE_LIST
"amd-gfx90a"
Expand All @@ -46,13 +49,13 @@ if(NOT __AOTRITON_INCLUDED)
"amd-gfx1250"
)
set(__AOTRITON_IMAGE_SHA256_LIST
"bb8bf2237b77fc503bc2967ea0d99d6ca419126c479e951ea42b712737128086" # amd-gfx90a
"f08edacf83c9ccf1c4bdcb51f1cab052d1680abea31c9e035f3f9fadb2f13ba4" # amd-gfx942
"307a37d729cda3a2120449909e5192cd71c2badccbd37f0222786098e69c7a91" # amd-gfx950
"c9cac7cf6f277168e1659ac2f04706f8823580b7c7e3e895f5a5503ed6bdd55f" # amd-gfx110x
"3177387a15c678b30057f4584d1fc1b8f8db56163890cb5c98f27450209f5a7b" # amd-gfx115x
"68572511ce6487a83f9014bd255bd69c8943f87d0c93bd57b2daac5fbc6c79c1" # amd-gfx120x
"c6ed084f1dce1c963c17055e38bcbc41d8e0fc48390d8fff6e11782454b26dbc" # amd-gfx1250
"a3d1a6868ce290ba8118618207093e785252eff4e18a64f495752cb5a03ffed6" # amd-gfx90a
"ccdbc7e3d96839be4895ee004f21531cc55d590c9018937b9e314bba363b3927" # amd-gfx942
"518fd072eb05948fc0a6c25a20832591c6406df865e3b691a2aeff3fd4c5ce1d" # amd-gfx950
"efe773e7a2c8adc995d90ecd0daca2db801285445ee10432df2b29c67d5b11d2" # amd-gfx110x
"1bc50e8aa8b6bda3410e92886ccca8fd45df3e60a6cbda9ffc58b2c541efd5c2" # amd-gfx115x
"6a465dbc03148bba8a2d78c4c2a3cb83155eca00f4f7f749e676402d7660968c" # amd-gfx120x
"4aaf71d6e510549d593757e5f88598df1e4a29cbcd91f70750ee8a76f65c027f" # amd-gfx1250
)
set(__AOTRITON_BASE_URL "$ENV{PYTORCH_AOTRITON_BASE_URL}")
if(NOT __AOTRITON_BASE_URL)
Expand Down
4 changes: 2 additions & 2 deletions related_commits
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ ubuntu|pytorch|torchvision|release/0.26|3d50b2155c02d26e78b9c09baf99f5c707d70215
centos|pytorch|torchvision|release/0.26|3d50b2155c02d26e78b9c09baf99f5c707d70215|https://github.com/ROCm/vision
ubuntu|pytorch|torchdata|release/0.11|377e64c1be69a9be6649d14c9e3664070323e464|https://github.com/pytorch/data
centos|pytorch|torchdata|release/0.11|377e64c1be69a9be6649d14c9e3664070323e464|https://github.com/pytorch/data
ubuntu|pytorch|torchaudio|release/2.11|34c52a67e8941bbd8e6adaca0eb0b9eabec11d78|https://github.com/pytorch/audio
centos|pytorch|torchaudio|release/2.11|34c52a67e8941bbd8e6adaca0eb0b9eabec11d78|https://github.com/pytorch/audio
ubuntu|pytorch|torchaudio|fix-ext-sources-str-2.11|c242809bedd7b99988858405c7aae6e1f8bd1c53|https://github.com/ethanwee1/audio
centos|pytorch|torchaudio|fix-ext-sources-str-2.11|c242809bedd7b99988858405c7aae6e1f8bd1c53|https://github.com/ethanwee1/audio
ubuntu|pytorch|ao|release/0.17.0|afb2844be99514f0d5ff42badd9c3ed0d1811d73|https://github.com/pytorch/ao
centos|pytorch|ao|release/0.17.0|afb2844be99514f0d5ff42badd9c3ed0d1811d73|https://github.com/pytorch/ao