Qianfeng 74d68e3b99 [CK_TILE] Simplify the codes in splitkv_combine pipeline (#1549) ...

* Simplify the codes in splitkv_combine pipeline

* Always set kPadSeqLenK=true for fmha splitkv kernels

* Change in Oacc Alignment and TileDistribution to be more adaptable to tile sizes

---------

Co-authored-by: Po Yen Chen <PoYen.Chen@amd.com>

2024-10-08 10:44:34 +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] FA bwd kernels optimization (#1397)

2024-08-16 13:40:10 -07:00

fmha_fwd_appendkv.py

[CK_TILE] Add PagedAttention kernels (#1387)

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

fmha_fwd_splitkv.py

[CK_TILE] Simplify the codes in splitkv_combine pipeline (#1549)

2024-10-08 10:44:34 +08:00

fmha_fwd.py

workaround rocm-6.2 compiler issue (#1421)

2024-07-31 16:03:59 +08:00