Mateusz Ozga
3c93d3c444
CK-Tile Grouped GEMM refactor and post PR fixes ( #1756 )
...
* Grouped gemm simple code refactor
* Offset invoker
* Invoke generic Run, and replace name of parrtitioner variable
* Tests fix type
* Removed namespaces
* Add template param to avoid implicit cast
* Remove generic function
* Constant value
* underline enum to int16_t
* Generalize partitioner function
* Remove whitespaces
* Rename function
* Using support
* Clang-format
* Clang-format
* Fn-partitioner description fn
* Typo
* Typo 2
* Better description
* Better description
* Refactor after review
* Use ctr instead of set fn
* Inovke ctr and typo
* Comments
* Remove unnecessary comment
* Review, remove modulo
2025-01-21 21:06:10 +01:00
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
