Skip to content

Pull requests: ROCm/TransformerEngine

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

Honor timeout settings in subprocess run wrapper ci-level 2 CI test level 2
#689 opened Aug 1, 2026 by ipanfilo Collaborator Loading…
1 of 13 tasks
Add pyyaml installation to ci prerequisites ci-level 3 CI test level 3
#688 opened Jul 31, 2026 by VeeraRajasekhar Contributor Loading…
13 tasks
@VeeraRajasekhar
3
Consolidating blockwise FP32 scale flag ci-level 3 CI test level 3
#687 opened Jul 31, 2026 by asdfvg123 Contributor Loading…
1 of 13 tasks
6
Updated AITER/QoLA ci-level 3 CI test level 3
#683 opened Jul 28, 2026 by Micky774 Contributor Loading…
13 tasks
3
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
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 ci-level 3 CI test level 3
#678 opened Jul 24, 2026 by shurale-nkn Loading…
2 of 13 tasks
18
microbenchmarks: add buffer rotation option
#677 opened Jul 23, 2026 by matthiasdiener Contributor Loading…
1 of 13 tasks
@matthiasdiener
10
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
1
[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
48
Updated CK/AITER Cmake Build ci-level 1 CI test level 1
#666 opened Jul 9, 2026 by Micky774 Contributor Loading…
13 tasks
10
Initial integration of a4w4 GEMM ci-level 3 CI test level 3
#663 opened Jul 7, 2026 by Micky774 Contributor • Draft
13 tasks
8
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
13
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
69
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.