carlushuang 851c3ed157 [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>

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

2.5 KiB

Raw Blame History

View Raw