Po Yen Chen 791802b381 [CK_TILE] fMHA batch_prefill block index & logits soft-capping optimizations (#2198) ...

* Write soft-sign in inline asm

* Change tile idx computation

* Add macro to turn off soft-sign asm opt

* Use simple for loop to avoid register spill

* Only do block id transform for masking cases

2025-05-16 15:14:46 +08:00

..

fmha_batch_prefill_kernel.hpp

[CK_TILE] fMHA batch_prefill block index & logits soft-capping optimizations (#2198)

2025-05-16 15:14:46 +08:00

fmha_bwd_kernel.hpp

Sync the kname with instance name (#1989)

2025-03-20 00:06:45 +08:00

fmha_fwd_appendkv_kernel.hpp

Remove using partitioner for all fmha kernels (#1778)

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

fmha_fwd_appendkv_tile_partitioner.hpp

[CK_TILE] Add PagedAttention kernels (#1387)

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

fmha_fwd_kernel.hpp

[CK_TILE] Add logits soft-capping & customization support to the FMHA forward kernel/pipelines (#2163)

2025-05-13 12:19:25 +08:00

fmha_fwd_splitkv_combine_kernel.hpp

Sync the kname with instance name (#1989)

2025-03-20 00:06:45 +08:00

fmha_fwd_splitkv_kernel.hpp

[CK_TILE] Add logits soft-capping & customization support to the FMHA forward kernel/pipelines (#2163)

2025-05-13 12:19:25 +08:00