Qianfeng ce2bdf42a9 Change in fwd-splitkv kernel to support num_splits=1 case (#1690) ...

* Change in fwd-splitkv kernel to support num_splits=1 case

* Update in codegen fwd-splitkv to make num_splits > 1 cases pass

* Specify instance traits in dispatch

* Fix link error for fp8 kernels

---------

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

2024-11-25 12:31:38 +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

Change in fwd-splitkv kernel to support num_splits=1 case (#1690)

2024-11-25 12:31:38 +08:00

fmha_fwd.py

[CK_TILE] Add fmha fwd headdim96 support (#1608)

2024-10-30 14:03:16 +08:00