Bartłomiej Kocot af66494880 [CK TILE] GEMM and Batched GEMM SplitK support (#1724) ...

* [CK TILE] Add split K support in GEMM

* Updates

* Fixes

* rebase

* fix

* Fix

* fixes

* support for batched gemm

2024-12-28 14:40:17 +01:00

..

gemm_pipeline_ag_bg_cr_base.hpp

[CK TILE] Add gemm compute pipeline v3 (#1661)

2024-11-28 17:51:49 +01:00

gemm_pipeline_ag_bg_cr_comp_v3.hpp

[CK TILE] GEMM and Batched GEMM SplitK support (#1724)

2024-12-28 14:40:17 +01:00

gemm_pipeline_ag_bg_cr_mem.hpp

[CK TILE] GEMM and Batched GEMM SplitK support (#1724)

2024-12-28 14:40:17 +01:00

gemm_pipeline_ag_bg_cr_scheduler.hpp

CK-Tile first draft of universal block gemm with interwave & intrawave scheduler (#1676)

2024-11-26 08:45:14 +01:00

gemm_pipeline_agmem_bgmem_creg_v1_default_policy.hpp

[CK TILE] GEMM and Batched GEMM SplitK support (#1724)

2024-12-28 14:40:17 +01:00

gemm_pipeline_agmem_bgmem_creg_v1.hpp

[CK TILE] GEMM and Batched GEMM SplitK support (#1724)

2024-12-28 14:40:17 +01:00

gemm_pipeline_agmem_bgmem_creg_v2_default_policy.hpp

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

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

gemm_pipeline_agmem_bgmem_creg_v2.hpp

[CK TILE] GEMM and Batched GEMM SplitK support (#1724)

2024-12-28 14:40:17 +01:00

gemm_pipeline_problem.hpp

CK-Tile first draft of universal block gemm with interwave & intrawave scheduler (#1676)

2024-11-26 08:45:14 +01:00

gemm_universal_pipeline_ag_bg_cr_policy.hpp

[CK TILE] GEMM and Batched GEMM SplitK support (#1724)

2024-12-28 14:40:17 +01:00

tile_gemm_shape.hpp

Ck tile gemm example (#1488)

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

tile_gemm_traits.hpp

[CK Tile] Improve the Layout, Padding, and Alignment features of CK Tile GEMM (#1651)

2024-11-11 18:08:25 -08:00