Skip to content

Add ROCm support for Torch 2.11#401

Merged
danieldk merged 6 commits intomainfrom
pytorch-2.11-rocm
Mar 26, 2026
Merged

Add ROCm support for Torch 2.11#401
danieldk merged 6 commits intomainfrom
pytorch-2.11-rocm

Conversation

@danieldk
Copy link
Member

Adds ROCm support for Torch 2.11. In order to do so:

  • Add ROCm 7.2.1.
  • Add aotriton 0.11.2b.
  • Remove old ROCm/aotriton versions.

@HuggingFaceDocBuilderDev

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
{
aotriton_0_10 = generic rec {
Copy link
Member

@sayakpaul sayakpaul Mar 26, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Do we not need it anymore?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

No, Torch 2.10 and 2.11 use aotriton 0.11.1 and 0.11.2 respectively.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Ah yeah. I see the modifications now.

sayakpaul
sayakpaul previously approved these changes Mar 26, 2026
Copy link
Member

@sayakpaul sayakpaul left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

All good pending the test failures.

@danieldk danieldk merged commit 953ef76 into main Mar 26, 2026
39 checks passed
@danieldk danieldk deleted the pytorch-2.11-rocm branch March 26, 2026 16:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants