Skip to content

Fix build on Pytorch 2.11 (#16505)#575

Merged
ipanfilo merged 1 commit into
devfrom
ipanfilo/pytorch_211_fix
May 5, 2026
Merged

Fix build on Pytorch 2.11 (#16505)#575
ipanfilo merged 1 commit into
devfrom
ipanfilo/pytorch_211_fix

Conversation

@ipanfilo
Copy link
Copy Markdown
Collaborator

@ipanfilo ipanfilo commented May 5, 2026

Description

Fix TE building on Pytorch 2.11

Fixes #16505

Type of change

  • Documentation change (change only to the documentation, either a fix or a new content)
  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • Infra/Build change
  • Code refactoring

Changes

  • Remove CUDAGuard hipification
  • Add manual selection of CUDAGuard/HIPGuardMasqueradingAsCUDA depending on Pytorch version

Checklist:

  • I have read and followed the contributing guidelines
  • The functionality is complete
  • I have commented my code, particularly in hard-to-understand areas
  • I have made corresponding changes to the documentation
  • My changes generate no new warnings
  • I have added tests that prove my fix is effective or that my feature works
  • New and existing unit tests pass locally with my changes

@ipanfilo ipanfilo requested a review from Micky774 May 5, 2026 05:33
@ipanfilo ipanfilo added the ci-level 1 CI test level 1 label May 5, 2026
Copy link
Copy Markdown
Collaborator

@wenchenvincent wenchenvincent left a comment

Choose a reason for hiding this comment

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

LGTM

@ipanfilo ipanfilo merged commit fcd5372 into dev May 5, 2026
10 checks passed
@ipanfilo ipanfilo deleted the ipanfilo/pytorch_211_fix branch May 5, 2026 17:46
ipanfilo added a commit that referenced this pull request May 7, 2026
Micky774 pushed a commit that referenced this pull request May 7, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ci-level 1 CI test level 1

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants