Skip to content

Pull requests: ROCm/TransformerEngine

New pull request New
21 Open 336 Closed
21 Open 336 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

[WIP] Speedup amax triton
#385 opened Nov 26, 2025 by matthiasdiener • Draft
13 tasks
@matthiasdiener
1
[DO NOT MERGE] Speedup amax kernel CI test
#384 opened Nov 26, 2025 by matthiasdiener • Draft
@matthiasdiener
Upcoming ROCm and JAX 0.8 support
#383 opened Nov 26, 2025 by ipanfilo Loading…
2 of 13 tasks
Enable AOTriton BWD V3 API
#382 opened Nov 25, 2025 by Micky774 Loading…
13 tasks
5
Old FP8 support code cleanup
#379 opened Nov 24, 2025 by ipanfilo Loading…
1 of 13 tasks
Re-enable supported GEMM configs
#378 opened Nov 24, 2025 by ipanfilo Loading…
13 tasks
Layernorm forward optimization
#377 opened Nov 24, 2025 by eliotwang Loading…
13 tasks
IFU dev v2.6
#374 opened Nov 19, 2025 by wangye805 Loading…
9 of 13 tasks
78
PyTorch FA test fix
#370 opened Nov 12, 2025 by Micky774 Loading…
13 tasks
1
Current scaling: two-stage amax kernel
#369 opened Nov 12, 2025 by matthiasdiener Loading…
3 of 15 tasks
@matthiasdiener
32
Userbuffer epic
#367 opened Nov 11, 2025 by alextmagro • Draft
AITER Native Padding Support and BSHD + Padding --> THD + Padding conversion
#354 opened Oct 28, 2025 by Micky774 Loading…
13 tasks
54
JAX FA Benchmarking Script
#351 opened Oct 24, 2025 by Micky774 Loading…
13 tasks
15
Fused Cross Entropy Triton - Loss Scaling and Vanishing Grads Bugfix
#336 opened Oct 16, 2025 by sarthak-amd Loading…
1 task done
7
[NO MERGE] Release v2.4 rocm
#334 opened Oct 8, 2025 by alextmagro Loading…
1
Triton norms dispatch refactor
#305 opened Sep 5, 2025 by Micky774 Loading…
13 tasks
12
heyi's layernorm optimization
#225 opened Jul 3, 2025 by eliotwang Loading…
8 of 13 tasks
@charyang-ai
Added Dockerfile for CI images
#195 opened May 28, 2025 by VeeraRajasekhar Loading…
7 of 13 tasks
30
[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! Type g p on any issue or pull request to go back to the pull request listing page.