Qianfeng 4e076909b6 Remove using partitioner for all fmha kernels (#1778) ...

* Remove using tile partitioner for fmha_fwd_kernel

* Remove using tile partitioner for fmha_fwd_splitkv and splitkv-combine kernels

* Remove using tile partitioner for fmha_fwd_appendkv kernel

* Unify the format of GetTileIndex

2024-12-29 14:29:56 +08: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] Use config name instead of data type in FmhaFwdTypeConfig<config> (#1731)

2024-12-10 11:36:18 +08:00

fmha_fwd_appendkv.py

Remove using partitioner for all fmha kernels (#1778)

2024-12-29 14:29:56 +08:00

fmha_fwd_splitkv.py

Remove using partitioner for all fmha kernels (#1778)

2024-12-29 14:29:56 +08:00

fmha_fwd.py

Remove using partitioner for all fmha kernels (#1778)

2024-12-29 14:29:56 +08:00