kylasa ab5d027866 Support for dtypes (fp8, bf8, bf16 and fp16) for the ck_tile/03_gemm example. (#1845) ...

* Support bf16/fb8/bf8 datatypes for ck_tile/gemm

* remove commented out code.

* Addressing code review comments and enabling universal_gemm for all the supported data types.

* Merge conflict resolution.

* Solve the memory pipeline compilation error. Merge with the new change of CShuffle

* finish the feature, pass the tests

* Fix the pipeline and add the benchmark script for other data types

---------

Co-authored-by: ThomasNing <thomas.ning@amd.com>

2025-02-07 15:07:06 -07:00

..

batched_gemm_kernel.hpp

[CK Tile] Spatially local GEMM tile partitioner. (#1843)

2025-01-31 00:10:16 +01:00

gemm_kernel.hpp

Support for dtypes (fp8, bf8, bf16 and fp16) for the ck_tile/03_gemm example. (#1845)

2025-02-07 15:07:06 -07:00

gemm_tile_partitioner.hpp

[CK Tile] Spatially local GEMM tile partitioner. (#1843)

2025-01-31 00:10:16 +01:00

grouped_gemm_kernel.hpp

[CK Tile] Spatially local GEMM tile partitioner. (#1843)

2025-01-31 00:10:16 +01:00