Skip to content

Backport fixes to TE2.10#534

Open
ipanfilo wants to merge 4 commits intorelease_v2.10_rocmfrom
ipanfilo/v210_backport
Open

Backport fixes to TE2.10#534
ipanfilo wants to merge 4 commits intorelease_v2.10_rocmfrom
ipanfilo/v210_backport

Conversation

@ipanfilo
Copy link
Copy Markdown
Collaborator

@ipanfilo ipanfilo commented Apr 9, 2026

Description

Backport fixes from dev:

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

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 and others added 4 commits April 9, 2026 15:35
@ipanfilo
Fix JAX sdist version pinning
25ddc63 
(cherry picked from commit c903b4e)
@ipanfilo
TE building over TheRock (#511)
ed84b92 
* Update Dockerfile to use ROCm TheRock
* Update wheels building script to work with ROCm TheRock and the latest Manylinux image
* Support default ROCm location /opt/rocm/core
* Fix UB code build on TheRock
* Support comma separated list of target GPU architectures
* Guess ROCm build from HIP_PLATFORM

(cherry picked from commit 67b93a8)
@ipanfilo
fix JAX distributed tests hang: ROCM-1719 (#483)
ba81fc1 
(cherry picked from commit e830722)
@sudhu2k @ipanfilo
Update CI runner in rocm-ci.yml to use linux-te-mi35x-8 (#504)
f6c33f6 
(cherry picked from commit 98ccd2e)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@AllenFarcas AllenFarcas Awaiting requested review from AllenFarcas

@VeeraRajasekhar VeeraRajasekhar Awaiting requested review from VeeraRajasekhar

@sudhu2k sudhu2k Awaiting requested review from sudhu2k

@wenchenvincent wenchenvincent Awaiting requested review from wenchenvincent wenchenvincent is a code owner

@wangye805 wangye805 Awaiting requested review from wangye805 wangye805 is a code owner

At least 1 approving review is required to merge this pull request.

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@ipanfilo @sudhu2k