ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 21:21:22 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
6091458300996a1b4a4f30ff25a828e8a40df7f2
composable_kernel/include/ck/tensor_operation/gpu
History
zjing14 6091458300 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>
2022-08-23 14:41:56 -05:00
..
block
Hotfix LDS data hazard in fused attention (#360)
2022-08-15 12:04:20 -05:00
device
Add examples of batched/grouped/SplitK Gemm for int8/bfp16/fp16/fp32 (#361)
2022-08-23 14:41:56 -05:00
element
Add example of Gemm + AddAddFastGelu (data type: int4) (#369)
2022-08-23 10:38:41 -05:00
grid
Add examples of batched/grouped/SplitK Gemm for int8/bfp16/fp16/fp32 (#361)
2022-08-23 14:41:56 -05:00
thread
Layernorm welford (#346)
2022-08-13 09:43:18 -05:00
warp
Hotfix LDS data hazard in fused attention (#360)
2022-08-15 12:04:20 -05:00