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
e26256cd7d8799fc2a12ea972a5de093233f0a4c
composable_kernel/include/ck
History
Shaojie WANG e26256cd7d 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>

[ROCm/composable_kernel commit: 45adb736e7]
2022-08-30 11:01:37 -05:00
..
host_utility
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
problem_transform
add license in file (#303)
2022-06-24 23:32:43 -05:00
tensor
add license in file (#303)
2022-06-24 23:32:43 -05:00
tensor_description
Fused attention (#345)
2022-08-13 00:16:14 -05:00
tensor_operation/gpu
Padding for attention: bmm+scale+softmax+bmm kernel (#385)
2022-08-30 11:01:37 -05:00
utility
Padding for attention: bmm+scale+softmax+bmm kernel (#385)
2022-08-30 11:01:37 -05:00
ck.hpp
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
stream_config.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00