composable_kernel/example/ck_tile/01_fmha/codegen at 36f241814e246b61cfeee798df4053f3fea9bd7d - composable_kernel - Public git mirror

ROCm/composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-08 23:38:11 +00:00

Files

History

Qianfeng Zhang fb496fa2d6 Add kQKHeaddimForGemmN and kVHeaddimForGemmN in order to support headdim 96

2024-10-12 13:27:02 +00:00

..

Add kQKHeaddimForGemmN and kVHeaddimForGemmN in order to support headdim 96

2024-10-12 13:27:02 +00:00

__init__.py

[CK_TILE] fmha forward split-kv + combine kernels (#1338 )

2024-06-26 17:41:15 +08:00

cmake_config.py

[CK_TILE] fmha forward split-kv + combine kernels (#1338 )

2024-06-26 17:41:15 +08:00

cpp_symbol_map.py

[CK_TILE] Add PagedAttention kernels (#1387 )

2024-08-28 20:50:43 +08:00