Adam Osewski 4f18c2ded3 Get hot loop and TailNum information before lunching kernel.

2024-10-01 14:40:03 +00: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

Early return if seqlen_k=0 on group mode (#1524)

2024-09-22 20:05:58 +08:00

gemm

Get hot loop and TailNum information before lunching kernel.

2024-10-01 14:40:03 +00:00

layernorm2d

layernorm2d forward (#1339)

2024-06-24 08:45:52 +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

Get hot loop and TailNum information before lunching kernel.

2024-10-01 14:40:03 +00: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