rocking 0d7c30f534 Merge branch 'develop' into ck_tile/rmsnorm

2024-10-29 18:53:00 +08: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_elementwise.hpp

host verification

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

reference_gemm.hpp

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

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

reference_im2col.hpp

[CK_TILE] Image to Column kernel (#1532)

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

reference_layernorm2d_fwd.hpp

update layernorm (#1570)

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

reference_permute.hpp

[CK_TILE] add generic_permute (#1607)

2024-10-29 18:05:53 +08:00

reference_reduce.hpp

Add reduce op

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

reference_rmsnorm2d_fwd.hpp

clang format

2024-10-24 17:05:36 +00:00

reference_rowwise_quantization2d.hpp

host verification

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

reference_softmax.hpp

topk_softmax (#1592)

2024-10-26 23:52:49 +08:00

reference_topk.hpp

topk_softmax (#1592)

2024-10-26 23:52:49 +08:00