Skip to content

Pull requests: ROCm/TransformerEngine

New pull request New
52 Open 598 Closed
52 Open 598 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

cherry-pick Update CI to use TheRock (#602)
#686 opened Jul 29, 2026 by VeeraRajasekhar Contributor Loading…
13 tasks
@VeeraRajasekhar
Updated AITER/QoLA ci-level 3 CI test level 3
#683 opened Jul 28, 2026 by Micky774 Contributor Loading…
13 tasks
2
Update CK-JIT to fix blob build failure if mv -no-clobber returns 1 ci-level 1 CI test level 1
#681 opened Jul 24, 2026 by ipanfilo Collaborator Loading…
1 of 13 tasks
gfx942 AITER V3 split-kv kernels ci-level 2 CI test level 2
#680 opened Jul 24, 2026 by ipanfilo Collaborator Loading…
13 tasks
2
microbenchmarks: usv implementation
#679 opened Jul 24, 2026 by matthiasdiener Contributor • Draft
13 tasks
[ROCm] Jax Add softmax sink (learnable off-by-one) support for the ROCm/CK fused attention backend
#678 opened Jul 24, 2026 by shurale-nkn Loading…
13 tasks
5
microbenchmarks: add buffer rotation option
#677 opened Jul 23, 2026 by matthiasdiener Contributor Loading…
1 of 13 tasks
@matthiasdiener
10
[WIP] Experimental FlyDSL GEMM backend for TE PyTorch (BF16/FP16/FP32/FP8/MXFP8) ci-level 1 CI test level 1
#676 opened Jul 22, 2026 by aris134 Contributor Loading…
@aris134
[wip] native nvfp4 GEMM support on gfx1250 ci-level 1 CI test level 1
#675 opened Jul 20, 2026 by matthiasdiener Contributor • Draft
13 tasks
@matthiasdiener
ci: bump te-rocm-wheels artifact retention 1d -> 5d
#673 opened Jul 17, 2026 by wenchenvincent Collaborator Loading…
2 tasks
3
[ROCm] Add THD/ragged bf16 (atomic16) backward test coverage
#671 opened Jul 16, 2026 by wenchenvincent Collaborator Loading…
10
[proof-of-concept] benchmarks dashboard
#670 opened Jul 14, 2026 by matthiasdiener Contributor • Draft
13 tasks
@matthiasdiener
Experimental Triton GEMM backend for TE PyTorch (BF16/FP16/FP32/FP8/MXFP8) ci-level 3 CI test level 3
#667 opened Jul 9, 2026 by wenchenvincent Collaborator Loading…
4 of 7 tasks
46
Updated CK/AITER Cmake Build ci-level 1 CI test level 1
#666 opened Jul 9, 2026 by Micky774 Contributor Loading…
13 tasks
9
Initial integration of a4w4 GEMM ci-level 3 CI test level 3
#663 opened Jul 7, 2026 by Micky774 Contributor • Draft
13 tasks
8
Grouped MXFP8 GEMMs with HipKittens ci-level 3 CI test level 3
#661 opened Jul 7, 2026 by alextmagro Contributor Loading…
47
CI: Fix runners GPU isolation ci-level 3 CI test level 3
#659 opened Jul 7, 2026 by leo-automation Collaborator Loading…
Ipanfilo/aiter split kv fwd ci-level 2 CI test level 2
#655 opened Jun 29, 2026 by ipanfilo Collaborator • Draft
13 tasks
[Feat] Added JAX-Triton bridge for ROCm
#649 opened Jun 24, 2026 by AllenFarcas Contributor Loading…
6 of 13 tasks
@AllenFarcas
7
Relax MXFP8 GEMM K constraint from multiple-of-128 to multiple-of-64
#642 opened Jun 19, 2026 by JohnQinAMD Loading…
13 tasks
1
grouped gemm microbenchmark: use te.GroupedLinear
#639 opened Jun 18, 2026 by matthiasdiener Contributor Loading…
13 tasks
@matthiasdiener
6
Interleaved Driver Benchmarking
#637 opened Jun 18, 2026 by Micky774 Contributor • Draft
13 tasks
4
[ROCm] Fix biased wgrad with fp32 gradient accumulation ci-level 1 CI test level 1
#634 opened Jun 18, 2026 by XinyuJiangCMU Loading…
3
Enable MultiCastTranspose for expert weights ci-level 3 CI test level 3
#628 opened Jun 16, 2026 by sudhu2k Contributor Loading…
8 of 13 tasks
@sudhu2k
1
Add ROCm HIP small-seq fused attention via crossattn_hip_kernel
#625 opened Jun 15, 2026 by VeeraRajasekhar Contributor Loading…
13 tasks
@VeeraRajasekhar
67
ProTip! What’s not been updated in a month: updated:<2026-06-29.