-
Notifications
You must be signed in to change notification settings - Fork 15
Pull requests: ROCm/TransformerEngine
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Remove filtering of supported cudagraph tests
#211
opened Jun 23, 2025 by
alextmagro
Loading…
4 of 13 tasks
Avoid poisoning process with JAX calls on TE import
#210
opened Jun 20, 2025 by
ipanfilo
Loading…
6 of 13 tasks
find dummy write can eliminate scratch in backward kernel
#192
opened May 23, 2025 by
eliotwang
Loading…
[ROCm] support triton-based flash-attn in TE
#177
opened May 1, 2025 by
wangye805
Loading…
8 of 13 tasks
Added fp8 gemm gelu_aux_bias support in release 1.9
#176
opened Apr 30, 2025 by
VeeraRajasekhar
Loading…
8 of 13 tasks
[Feat] optim TE_HIPBLASLT_ALGO_LOAD logic
#156
opened Mar 24, 2025 by
RuibinCheung
Loading…
13 tasks
Update attention example attention.ipynb
#152
opened Mar 19, 2025 by
anhminhnguyenhoang
Loading…
5 of 13 tasks
[ROCm] enable hdim=64 and packed qkv layout for ck bwd v3 kernels
#139
opened Mar 5, 2025 by
wangye805
Loading…
8 of 13 tasks
Honor the NVTE_FUSED_ATTN_<backend> in test_fused_attn.py
#123
opened Feb 11, 2025 by
wangye805
Loading…
13 tasks
ProTip!
Updated in the last three days: updated:>2025-06-21.