Skip to content

Pull requests: ROCm/TransformerEngine

New pull request New
23 Open 411 Closed
23 Open 411 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add ROCm mismatch bypass env flag
#466 opened Feb 28, 2026 by jiagaoxiang Loading…
3 tasks
8
Release 2.6 fix: cuda graph dropout accuracy
#463 opened Feb 25, 2026 by Micky774 Loading…
2 of 13 tasks
6
[NO MERGE] Integrate CK varlen cross attention for small-seq (s_q=1, s_kv<=16)
#461 opened Feb 24, 2026 by VeeraRajasekhar Loading…
13 tasks
@VeeraRajasekhar
61
IFU v2.10
#460 opened Feb 24, 2026 by alextmagro Loading…
49
Always use V2 hipify. Make all hipify results consistent
#459 opened Feb 24, 2026 by ipanfilo Loading…
2 of 13 tasks
Update CI to label-driven testing
#450 opened Feb 17, 2026 by Micky774 Loading…
1 of 13 tasks
24
Added initial GH Copilot instructions
#448 opened Feb 12, 2026 by Micky774 Loading…
2 of 13 tasks
Update AITER subcommit and refactor internal AITER/CK FA API usage
#446 opened Feb 9, 2026 by Micky774 Loading…
13 tasks
64
Support release wheels with local version
#444 opened Feb 6, 2026 by ipanfilo Loading…
1 of 13 tasks
Make distinctive ROCm TE wheels names
#441 opened Feb 3, 2026 by ipanfilo Loading…
1 of 13 tasks
2
Update README.rst
#435 opened Jan 28, 2026 by aris134 • Draft
1 of 13 tasks
Grouped GEMM with ck_tile
#434 opened Jan 28, 2026 by matthiasdiener Loading…
6 of 18 tasks
@matthiasdiener
77
release v2.8 rocm
#428 opened Jan 22, 2026 by VeeraRajasekhar Loading…
13 tasks
@VeeraRajasekhar
24
MXFP4 Cast Transpose Triton [WIP]
#422 opened Jan 20, 2026 by sarthak-amd • Draft
1 task done
19
Hotfix/fused ce triton
#409 opened Jan 12, 2026 by sarthak-amd Loading…
7
CI: Switch GHA pipeline to build and test wheels
#400 opened Dec 9, 2025 by leo-amd • Draft
1 of 6 tasks
2
Layernorm forward optimization
#377 opened Nov 24, 2025 by eliotwang Loading…
13 tasks
Userbuffer epic
#367 opened Nov 11, 2025 by alextmagro Loading…
33
Fused Cross Entropy Triton - Loss Scaling and Vanishing Grads Bugfix
#336 opened Oct 16, 2025 by sarthak-amd Loading…
1 task done
8
heyi's layernorm optimization
#225 opened Jul 3, 2025 by eliotwang Loading…
8 of 13 tasks
@charyang-ai
[ROCm] support triton-based flash-attn in TE
#177 opened May 1, 2025 by wangye805 Loading…
8 of 13 tasks
6
Update attention example attention.ipynb
#152 opened Mar 19, 2025 by anhminhnguyenhoang Loading…
5 of 13 tasks
1
Honor the NVTE_FUSED_ATTN_<backend> in test_fused_attn.py
#123 opened Feb 11, 2025 by wangye805 Loading…
13 tasks
3
ProTip! What’s not been updated in a month: updated:<2026-01-28.