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 |
|