ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
54af9c3aa069cbb328d752ca7ec9084fa8f47de5
composable_kernel/example/35_splitK_gemm
History
zjing14 1401b574e0 Add examples of batched/grouped/SplitK Gemm for int8/bfp16/fp16/fp32 (#361) 
* add examples into grouped/batched_gemm

* adding splitK examples

* fixed splitK

* add bfp16 int8 example into splitK

* formatting

* use static_cast

* added common for batched_gemm

* add commons for examples of splitK/batched/grouped_gemm

* return true

* adjust splitK check tol

* update example

Co-authored-by: Chao Liu <lc.roy86@gmail.com>

[ROCm/composable_kernel commit: 6091458300]
2022-08-23 14:41:56 -05:00
..
CMakeLists.txt
Add examples of batched/grouped/SplitK Gemm for int8/bfp16/fp16/fp32 (#361)
2022-08-23 14:41:56 -05:00
run_splitK_gemm_example.inc
Add examples of batched/grouped/SplitK Gemm for int8/bfp16/fp16/fp32 (#361)
2022-08-23 14:41:56 -05:00
splitK_gemm_xdl_bfp16.cpp
Add examples of batched/grouped/SplitK Gemm for int8/bfp16/fp16/fp32 (#361)
2022-08-23 14:41:56 -05:00
splitK_gemm_xdl_fp16.cpp
Add examples of batched/grouped/SplitK Gemm for int8/bfp16/fp16/fp32 (#361)
2022-08-23 14:41:56 -05:00
splitK_gemm_xdl_fp32.cpp
Add examples of batched/grouped/SplitK Gemm for int8/bfp16/fp16/fp32 (#361)
2022-08-23 14:41:56 -05:00
splitK_gemm_xdl_int8.cpp
Add examples of batched/grouped/SplitK Gemm for int8/bfp16/fp16/fp32 (#361)
2022-08-23 14:41:56 -05:00