ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
851c3ed1576fed2593b824fb83b3f9349f6aeb09
composable_kernel/test/position_embedding
History
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
..
CMakeLists.txt
[CK_TILE] support alibi (#1269)
2024-05-07 22:32:54 +08:00
position_embedding.cpp
[CK_TILE] support alibi (#1269)
2024-05-07 22:32:54 +08:00