Thomas Ning caacd38830 Ck tile gemm example (#1488) ...

* Checkpoint: Finished with the tile example & kernel verification, working on the different matrix layout

* Finished the Matrix Layout feature set up. Note: Need to modify the inner block to solve the shuffle problem in the future.

* Fix: Clang Format, API fixed from fmha

* fix with better naming convention

* revert back the pipeline code of fmha

* Fixed: Addressed the comments and merge the GEMM shape of GEMM Operator and FMHA Operator to one.

* clang format with the reference_gemm file

* convert the clang format with the remod.py

* Changed the format and variable name of the kernel gemm_shape and partitioner

---------

Co-authored-by: thomasning <thomasning@banff-cyxtera-s70-4.ctr.dcgpu>

2024-09-07 16:23:32 +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_gemm.hpp

Ck tile gemm example (#1488)

2024-09-07 16:23:32 +08: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