ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Enable grouped conv bwd wei bf16 NGCHW (#1589)

Browse Source 
* Enable grouped conv bwd wei bf16 NGCHW

* fixes

* fixes

* Fixes

* fixes

* fixes

* Fixes
This commit is contained in:
Bartłomiej Kocot
2024-10-22 16:18:28 +02:00
committed by GitHub
parent 0394f8a713
commit 82fc53835a
28 changed files with 667 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
script/convert_miopen_driver_to_profiler.py
View File

@@ -65,8 +65,9 @@ def parse_data_type(args):
if args.ck_profier_op == "grouped_conv_fwd":
args.data_type = 3
if args.data_type == "bfp16":
if args.ck_profier_op == "grouped_conv_bwd_weight" or \
args.ck_profier_op == "grouped_conv_bwd_data" or \
if args.ck_profier_op == "grouped_conv_bwd_weight":
args.data_type = 5
if args.ck_profier_op == "grouped_conv_bwd_data" or \
args.ck_profier_op == "grouped_conv_fwd":
args.data_type = 2