composable_kernel/example/ck_tile/01_fmha/codegen/ops at b34ddf5f714cf712bc33b609c4be2daff3071c33 - 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 b34ddf5f71 Merge remote-tracking branch 'origin/feature/cond-add-splitkv' into feature/fmha-fwd-appendkv

2024-07-12 04:42:45 +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

Fix wrong K values after appending

2024-06-25 10:12:13 +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