This website requires JavaScript.
Explore
Help
Register
Sign In
ROCm
/
composable_kernel
Watch
1
Star
0
Fork
0
You've already forked composable_kernel
mirror of
https://github.com/ROCm/composable_kernel.git
synced
2026-07-13 10:37:42 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
4141cf1a97a7e6d89cd6176c863323bdc43df3df
composable_kernel
/
example
/
ck_tile
/
01_fmha
/
codegen
/
ops
History
zanzhang
4141cf1a97
Revert "add chunked prefill into mha varlen"
...
This reverts commit
faf9be7de6
.
2025-05-21 12:48:01 +08:00
..
__init__.py
[CK_TILE] fmha forward split-kv + combine kernels (
#1338
)
2024-06-26 17:41:15 +08:00
fmha_batch_prefill.py
Use only qr_async pipeline for batch_prefill (
#2195
)
2025-05-15 11:47:29 -07:00
fmha_bwd.py
[CK_TILE] Add logits soft-capping & customization support to the FMHA forward kernel/pipelines (
#2163
)
2025-05-13 12:19:25 +08:00
fmha_fwd_appendkv.py
Only generate specific hdim (
#2120
)
2025-04-24 18:52:58 +08:00
fmha_fwd_splitkv.py
Merge remote-tracking branch 'origin/fix_fa_codegen' into Fa_seqlen1_pass
2025-05-16 14:23:16 +08:00
fmha_fwd.py
Revert "add chunked prefill into mha varlen"
2025-05-21 12:48:01 +08:00
