ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
e7ff49610b8bc5c4cdca03bffe6a0f5f18226e14
composable_kernel/include/ck_tile/ops
History
Qianfeng 99c360c6e4 [CK_TILE] Simplify the codes in splitkv_combine pipeline (#1549) 
* Simplify the codes in splitkv_combine pipeline

* Always set kPadSeqLenK=true for fmha splitkv kernels

* Change in Oacc Alignment and TileDistribution to be more adaptable to tile sizes

---------

Co-authored-by: Po Yen Chen <PoYen.Chen@amd.com>

[ROCm/composable_kernel commit: 74d68e3b99]
2024-10-08 10:44:34 +08:00
..
common
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
epilogue
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
fmha
[CK_TILE] Simplify the codes in splitkv_combine pipeline (#1549)
2024-10-08 10:44:34 +08:00
gemm
Ck tile gemm padding dim (#1516)
2024-09-18 11:32:29 -07:00
image_to_column
[CK_TILE] Image to Column kernel (#1532)
2024-09-27 22:57:38 +02:00
layernorm2d
[Ck tile] Support layernorm one pass (#1512)
2024-10-07 14:25:53 +08:00
reduce/block
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
welford
layernorm2d forward (#1339)
2024-06-24 08:45:52 +08:00
common.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
epilogue.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
fmha.hpp
[CK_TILE] Add PagedAttention kernels (#1387)
2024-08-28 20:50:43 +08:00
gemm.hpp
Ck tile gemm example (#1488)
2024-09-07 16:23:32 +08:00
image_to_column.hpp
[CK_TILE] add missing vector header (#1537)
2024-10-01 07:58:20 -07:00
layernorm2d.hpp
layernorm2d forward (#1339)
2024-06-24 08:45:52 +08:00
reduce.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
welford.hpp
layernorm2d forward (#1339)
2024-06-24 08:45:52 +08:00