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
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
jakpiase
feb9a2bd9b
Add IsSupportedArgument() to gemm_kernel ( #1698 )
...
* add IsSupportedArgument to gemm_kernel
* add ut and do some refactoring
* switched to ck_tile's integral_constant
2024-12-05 09:02:13 +01:00
Bartłomiej Kocot
f49b595dc0
[CK TILE] Add gemm compute pipeline v3 ( #1661 )
...
* [CK TILE] Add gemm compute pipeline v3
* Enable universal gemm compute pipeline.
* Rename example and add compute pipeline.
* Introduce ag bg cr pipeline impl base.
* Refactor to reuse code.
* Cleaning
* Formatting.
---------
Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com >
Co-authored-by: Adam Osewski <Adam.Osewski@amd.com >
2024-11-28 17:51:49 +01:00
