Thomas Ning 5d671a5fc4 CK Tile GEMM CICD fixed & register block method refactor (#1776) ...

* refactor the block_gemm_areg_breg_creg_v1 and add the v2 policy with 2x2 warp gemm

* Finished the 2x2 warp gemm policy and the block selection mechanism

* Clang format

* address poyen's comment

* Address feedbacks

* Fixed the compilation issue

* Change the function name

2025-01-13 13:10:44 +08:00

..

block_gemm_areg_bgmem_creg_v1_default_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_gemm_areg_bgmem_creg_v1.hpp

[CK-Tile] Universal gemm memory bound pipeline (#1558)

2024-10-30 10:05:15 +01:00

block_gemm_areg_breg_creg_v1_custom_policy.hpp

[CK_TILE] FA bwd kernels optimization (#1397)

2024-08-16 13:40:10 -07:00

block_gemm_areg_breg_creg_v1_default_policy.hpp

[CK_TILE] FA bwd kernels optimization (#1397)

2024-08-16 13:40:10 -07:00

block_gemm_areg_breg_creg_v1.hpp

CK Tile GEMM CICD fixed & register block method refactor (#1776)

2025-01-13 13:10:44 +08:00

block_gemm_areg_bsmem_creg_one_warp_v1.hpp

[CK_TILE] Add fmha fwd N-Warp S-Shuffle pipeline (fmha fwd splitkv pipeline variant) (#1705)

2024-12-20 14:41:01 +08:00

block_gemm_areg_bsmem_creg_v1_custom_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_gemm_areg_bsmem_creg_v1_default_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_gemm_areg_bsmem_creg_v1.hpp

[CK_TILE] More fmha splitkv optimizations (#1588)

2024-10-26 18:35:45 +08:00

block_gemm_areg_bsmem_creg_v2_custom_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_gemm_areg_bsmem_creg_v2_default_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_gemm_areg_bsmem_creg_v2.hpp

[CK_TILE] Add fmha fwd N-Warp S-Shuffle pipeline (fmha fwd splitkv pipeline variant) (#1705)

2024-12-20 14:41:01 +08:00

block_gemm_asmem_breg_creg_v1_custom_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_gemm_asmem_breg_creg_v1_default_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_gemm_asmem_breg_creg_v1.hpp

[CK_TILE] More fmha splitkv optimizations (#1588)

2024-10-26 18:35:45 +08:00

block_gemm_asmem_bsmem_creg_v1_custom_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_gemm_asmem_bsmem_creg_v1_default_policy.hpp

Ck tile gemm example (#1488)

2024-09-07 16:23:32 +08:00

block_gemm_asmem_bsmem_creg_v1.hpp

[CK-Tile] Universal gemm memory bound pipeline (#1558)

2024-10-30 10:05:15 +01:00

block_gemm_problem.hpp

CK Tile FA Training kernels (#1286)

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

block_universal_gemm_as_bs_cr.hpp

[CK TILE] Fix universal gemm template keywords (#1704)

2024-11-29 20:51:09 -08:00