composable_kernel/example/ck_tile/01_fmha/codegen/ops at ca4b208b601094bba588d2a1be6e53ee3cfcf5df - composable_kernel - Public git mirror

ROCm/composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-07 00:04:37 +00:00

Files

History

PoYen, Chen fffd6799e6 Instantiate multiple kernels for RoPE approaches

2024-07-20 02:28:21 +00:00

..

__init__.py

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

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

fmha_bwd.py

[CK_TILE] wa prec, remove sgpr offset for inline asm (#1356 )

2024-07-08 11:09:55 -07:00

fmha_fwd_appendkv.py

Instantiate multiple kernels for RoPE approaches

2024-07-20 02:28:21 +00:00

fmha_fwd_splitkv.py

[CK_TILE] wa prec, remove sgpr offset for inline asm (#1356 )

2024-07-08 11:09:55 -07:00

fmha_fwd.py

[CK_TILE] wa prec, remove sgpr offset for inline asm (#1356 )

2024-07-08 11:09:55 -07:00