Adam Osewski 8bb2bb4a05 Grouped Gemm + SplitK + simplified Kernel Args (#669) ...

* simplify karg in device/grid split-k op

* fix mk_kn_mn instances

* add more instances

* B2C with 3D grid for KSplit

* Remove unused code.

* Use default B2C (3D grid) in grid gemm v2r4r2.

* Device gemm splitk use B2C map.

* Device GroupedGemmXdlSplitKCShuffle

* Example for GroupedGemm Xdl SplitK

* Introduce Device GroupedGemmSplitK

* Fix updating kbatch size.

* Add instance mk-nk-mn

* Enable set kbatch in profiler.

* Add GGemmSplitK mk-kn-mn instances

* Add more instances & split into multiple files.

* minor fix

* tuning

* clean

* disabled failed instances

* use pipe v2

* Ignore arg on not supported arch.

* fix warning

---------

Co-authored-by: carlushuang <carlus.huang@amd.com>
Co-authored-by: Adam Osewski <aosewski@amd.com>
Co-authored-by: zjing14 <zhangjing14@gmail.com>
Co-authored-by: Jing Zhang <jizhan@amd.com>
Co-authored-by: root <root@ctr-ubbsmc15.amd.com>

2023-04-24 15:43:36 -05:00

..

include/ck/library

Grouped Gemm + SplitK + simplified Kernel Args (#669)

2023-04-24 15:43:36 -05:00

src

Grouped Gemm + SplitK + simplified Kernel Args (#669)

2023-04-24 15:43:36 -05:00

CMakeLists.txt

Clean up conv example, Instances, profiler and test (#324)

2022-07-29 18:19:25 -05:00