ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-04 05:31:24 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
3da5c19e629174c234fe86c17ebd04732ea548b7
composable_kernel/include/ck/tensor_operation/gpu/element
History
Shaojie WANG 45adb736e7 Padding for attention: bmm+scale+softmax+bmm kernel (#385) 
* add padding algo for bmm+scale+softmax+bmm. Version for verification

* remove verification code

* remove comments

* add padded bmm scale softmax bmm example

* format

* refactor

* add comments for usages of padding bmm+scale+softmax+bmm

Co-authored-by: Chao Liu <lc.roy86@gmail.com>
2022-08-30 11:01:37 -05:00
..
binary_element_wise_operation.hpp
CGEMM examples bf16, fp32, int8 (#332)
2022-08-02 14:52:27 -05:00
element_wise_operation.hpp
Add int4 example for convnd_fwd_bias_relu_add (#375)
2022-08-25 17:08:43 -05:00
unary_element_wise_operation.hpp
Padding for attention: bmm+scale+softmax+bmm kernel (#385)
2022-08-30 11:01:37 -05:00