ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-03 13:48:30 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
d563dac4249b54d7bbcfb3a2e78cd0bfb37becf8
composable_kernel/include/ck_tile/ops/fmha
History
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
..
block
[CK_TILE] fMHA batch_prefill block index & logits soft-capping optimizations (#2198)
2025-05-16 15:14:46 +08:00
kernel
[CK_TILE] fMHA batch_prefill block index & logits soft-capping optimizations (#2198)
2025-05-16 15:14:46 +08:00
pipeline
[CK_TILE] fMHA batch_prefill block index & logits soft-capping optimizations (#2198)
2025-05-16 15:14:46 +08:00