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
7ed9e9a34870f0cb4f4047b81480ca406b997088
composable_kernel/include
History
Shaojie WANG 7ed9e9a348 example of conv bwd weight 1d/2d/3d fp32/fp16/bf16 xdl (#244) 
* enable example of conv 1d/3d for bwd weight

* make bf16 kernel do not use atomic add

* using new gridwise gemm for bwd weight on convnd bwd weight

Co-authored-by: Chao Liu <chao.liu2@amd.com>

[ROCm/composable_kernel commit: ac543313bf]
2022-05-20 17:20:10 -05:00
..
ck
example of conv bwd weight 1d/2d/3d fp32/fp16/bf16 xdl (#244)
2022-05-20 17:20:10 -05:00