ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 11:47:48 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
bd8024b84ab72d2fee145e24e5eb94e2501a792e
composable_kernel/client_example/20_splitk_gemm
History
Rostyslav Geyyer eac50708d9 Add instances/ckProfiler/client example for fp8/fp16 mixed precision Gemm (#853) 
* Add ComputeType arg to splitk device and gridwise ops

* Update for gridwise op compatibility

* Update bf16 and int8 splitk gemm examples with ComputeType

* Add instances

* Update ckProfiler for mixed precision cases

* Add a mixed precision splitK gemm client example

---------

Co-authored-by: zjing14 <zhangjing14@gmail.com>
2023-08-22 09:34:49 -05:00
..
CMakeLists.txt
Add instances/ckProfiler/client example for fp8/fp16 mixed precision Gemm (#853)
2023-08-22 09:34:49 -05:00
splitK_gemm_fp16_f8.cpp
Add instances/ckProfiler/client example for fp8/fp16 mixed precision Gemm (#853)
2023-08-22 09:34:49 -05:00