ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
47958ebd075f5a110ed3fa77f2dd3253133cd883
composable_kernel/client_example/20_splitk_gemm
History
Rostyslav Geyyer 6f9eeb3190 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>

[ROCm/composable_kernel commit: eac50708d9]
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