carlushuang 9be788a9aa [CK_TILE] support alibi (#1269) ...

* add alibi support

* fix code

* update code based on comment

* Support more hdim

* fix fp8 bias

* support seqlen_k=0 case

* remove unused printf

* fix format

---------

Co-authored-by: rocking <ChunYu.Lai@amd.com>

[ROCm/composable_kernel commit: 851c3ed157]

2024-05-07 22:32:54 +08:00

12 KiB

Raw Blame History

View Raw