rocking 826ee18a11 Add reduce op

2024-10-25 22:51:15 +00:00

..

add_rmsnorm2d_rdquant

Add reduce op

2024-10-25 22:51:15 +00:00

common

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

epilogue

Ck tile gemm cshuffle & CK Tile GEMM restructure (#1535)

2024-10-10 18:02:22 +08:00

fmha

[CK_TILE] Optimize fmha splitkv & splitkv combine kernels (#1577)

2024-10-21 10:52:11 +08:00

gemm

[CK_TILE] Add block universal gemm pipeline policy (#1557)

2024-10-15 13:53:41 +02:00

image_to_column

[CK_TILE] Image to Column kernel (#1532)

2024-09-27 22:57:38 +02:00

layernorm2d

Refine pipeline name

2024-10-24 20:42:40 +00:00

reduce/block

update layernorm (#1570)

2024-10-22 09:26:18 +08:00

reduce2d/block

Remove static assert to prevent compile fail

2024-10-24 06:09:23 +00:00

rmsnorm2d

Add reduce op

2024-10-25 22:51:15 +00:00

welford

update layernorm (#1570)

2024-10-22 09:26:18 +08:00

add_rmsnorm2d_rdquant.hpp

Add add_rmsnorm2d_rdquant kernel

2024-10-25 20:50:48 +00:00

common.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

epilogue.hpp

Ck tile gemm cshuffle & CK Tile GEMM restructure (#1535)

2024-10-10 18:02:22 +08:00

fmha.hpp

[CK_TILE] Add PagedAttention kernels (#1387)

2024-08-28 20:50:43 +08:00

gemm.hpp

[CK_TILE] Optimize fmha splitkv & splitkv combine kernels (#1577)

2024-10-21 10:52:11 +08:00

image_to_column.hpp

[CK_TILE] add missing vector header (#1537)

2024-10-01 07:58:20 -07:00

layernorm2d.hpp

update layernorm (#1570)

2024-10-22 09:26:18 +08:00

reduce.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

rmsnorm2d.hpp

Add rmsnorm2d

2024-10-23 19:23:51 +00:00

welford.hpp

update layernorm (#1570)

2024-10-22 09:26:18 +08:00