ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-01 12:11:19 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
5948dbffe4d0bbe4d1802a047bd8599ba662386e
composable_kernel/example/ck_tile/01_fmha/codegen/ops
History
rocking 5948dbffe4 Support fp8 dynamic quantization for fmha (#3206) 
* Support qscale for dynamic quant, remove static quant

* Support hdim=256

* Remove bias test case for fp8

---------

Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>
Co-authored-by: asleepzzz <hanwen.chang@amd.com>
2025-11-24 16:28:25 +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
[CK_TILE] Support WMMA (gfx12) in FMHA (#2528)
2025-10-29 13:31:08 -07:00
fmha_bwd.py
[CK_TILE] Support WMMA (gfx12) in FMHA (#2528)
2025-10-29 13:31:08 -07:00
fmha_fwd_appendkv.py
[CK_TILE] Support WMMA (gfx12) in FMHA (#2528)
2025-10-29 13:31:08 -07:00
fmha_fwd_splitkv.py
Revert "Add attn sink (#2892)" (#3250)
2025-11-20 07:55:15 -08:00
fmha_fwd.py
Support fp8 dynamic quantization for fmha (#3206)
2025-11-24 16:28:25 +08:00
fmha_pagedkv_prefill.py
Revert "Add attn sink (#2892)" (#3250)
2025-11-20 07:55:15 -08:00