ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
2655746971424cfb6ca84ea6088743799dbf3be8
composable_kernel/include
History
Po Yen Chen 2655746971 [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

[ROCm/composable_kernel commit: 791802b381]
2025-05-16 15:14:46 +08:00
..
ck
Update the buffer load/store intrinsic names for clang>=20. (#2192)
2025-05-13 10:18:14 -07:00
ck_tile
[CK_TILE] fMHA batch_prefill block index & logits soft-capping optimizations (#2198)
2025-05-16 15:14:46 +08:00