composable_kernel/include/ck_tile/host/reference at 4cf45f1b9c45b18138e3e647687f87d28f59d313 - composable_kernel - Public git mirror

ROCm/composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-15 03:30:11 +00:00

Files

History

Adam Osewski 8bd49370f1 Refactoring & Move Layout info to pipeline problem.

2024-10-01 14:38:19 +00:00

..

reference_batched_dropout.hpp

CK Tile FA Training kernels (#1286 )

2024-06-04 13:12:45 -05:00

reference_batched_elementwise.hpp

introducing ck_tile! (#1216 )

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

reference_batched_gemm.hpp

introducing ck_tile! (#1216 )

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

reference_batched_masking.hpp

introducing ck_tile! (#1216 )

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

reference_batched_rotary_position_embedding.hpp

[CK_TILE] Add PagedAttention kernels (#1387 )

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

reference_batched_softmax.hpp

introducing ck_tile! (#1216 )

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

reference_gemm.hpp

Refactoring & Move Layout info to pipeline problem.

2024-10-01 14:38:19 +00:00

reference_im2col.hpp

introducing ck_tile! (#1216 )

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

reference_layernorm2d.hpp

layernorm2d forward (#1339 )

2024-06-24 08:45:52 +08:00

reference_reduce.hpp

introducing ck_tile! (#1216 )

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

reference_softmax.hpp

introducing ck_tile! (#1216 )

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