Add support for NGCHW in grouped conv fwd (#1499)

* Support NGCHW in grouped conv fwd

* Remove not needed variable

* Fixes

[ROCm/composable_kernel commit: 4ba52b35dc]
This commit is contained in:
Bartłomiej Kocot
2024-09-20 10:45:46 +02:00
committed by GitHub
parent f6c6c375db
commit e4f4e04add
27 changed files with 1620 additions and 305 deletions

View File

@@ -28,6 +28,8 @@ def parse_layouts(args):
args.in_layout == "NCDHW":
if args.ck_profier_op == "grouped_conv_bwd_weight":
args.layout = 3
elif args.ck_profier_op == "grouped_conv_fwd":
args.layout = 2
else:
print('Not supported layout for this op')
exit(1)