Skip to content

Commit f18fa12

Browse files
kmilanovicTTvobojevicTTvbrkicTT
authored
Tests for convTranspose2d operator (#1193)
Tests for convTranspose2d Closes [#888](#888) --------- Co-authored-by: Vladica Obojevic <[email protected]> Co-authored-by: Vladimir Brkic <[email protected]>
1 parent f243c42 commit f18fa12

15 files changed

+1661
-6
lines changed

‎forge/forge/op_repo/pytorch_operators.py‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,12 @@
4040
OperatorParamNumber("padding", int, 1, 1),
4141
],
4242
),
43+
OperatorDefinition(
44+
"conv_transpose_2d",
45+
"torch.nn.ConvTranspose2d",
46+
1,
47+
instantiate=True,
48+
),
4349
# Unary operators (implemented)
4450
OperatorDefinition("relu", "torch.relu", 1),
4551
OperatorDefinition("sqrt", "torch.sqrt", 1),

‎forge/test/operators/pytorch/ids/failing_reasons_register.py‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,14 @@ class FailingReasonsRegister:
2222
("conv2d", FailingReasons.DATA_MISMATCH),
2323
("conv2d", FailingReasons.NOT_IMPLEMENTED),
2424
("conv2d", FailingReasons.UNSUPPORTED_SPECIAL_CASE),
25+
("conv_transpose_2d", FailingReasons.ALLOCATION_CIRCULAR_BUFFER),
26+
("conv_transpose_2d", FailingReasons.ALLOCATION_FAILED),
27+
("conv_transpose_2d", FailingReasons.ASSERT_DIM),
28+
("conv_transpose_2d", FailingReasons.ASSERT_GROUPS),
29+
("conv_transpose_2d", FailingReasons.ASSERT_STRIDE),
30+
("conv_transpose_2d", FailingReasons.DATA_MISMATCH),
31+
("conv_transpose_2d", FailingReasons.MLIR_RUNTIME),
32+
("conv_transpose_2d", FailingReasons.SPECIAL_VALUES),
2533
("cumsum", FailingReasons.DATA_MISMATCH),
2634
("div", FailingReasons.DATA_MISMATCH),
2735
("div", FailingReasons.DTYPE_MISMATCH),

‎forge/test/operators/pytorch/ids/push.txt‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
# xfail_conv_transpose_2d_ASSERT_DIM ids
2+
conv_transpose_2d-FROM_ANOTHER_OP-{'in_channels': 11, 'out_channels': 11, 'kernel_size': (21, 2), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': (2, 15), 'dtype': None}-(1, 11, 45, 17)-None-None
3+
4+
5+
# xfail_conv_transpose_2d_ASSERT_GROUPS ids
6+
conv_transpose_2d-FROM_ANOTHER_OP-{'in_channels': 32, 'out_channels': 38, 'kernel_size': (9, 9), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 2, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 32, 32, 64)-None-None
7+
8+
9+
# xfail_conv_transpose_2d_ASSERT_STRIDE ids
10+
conv_transpose_2d-FROM_ANOTHER_OP-{'in_channels': 11, 'out_channels': 11, 'kernel_size': (14, 5), 'stride': (6, 2), 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 11, 45, 17)-None-None

‎forge/test/operators/pytorch/ids/xfail/conv_transpose_2d_allocation_circular_buffer.txt‎

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
# xfail_conv_transpose_2d_ALLOCATION_CIRCULAR_BUFFER ids
2+
3+
conv_transpose_2d-FROM_ANOTHER_OP-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
4+
conv_transpose_2d-FROM_ANOTHER_OP-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None
5+
conv_transpose_2d-FROM_HOST-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
6+
conv_transpose_2d-FROM_HOST-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None
7+
conv_transpose_2d-CONST_EVAL_PASS-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
8+
conv_transpose_2d-CONST_EVAL_PASS-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None
9+
conv_transpose_2d-FROM_ANOTHER_OP-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 2, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
10+
conv_transpose_2d-FROM_ANOTHER_OP-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 3, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None
11+
conv_transpose_2d-FROM_HOST-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 2, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
12+
conv_transpose_2d-FROM_HOST-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 3, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None
13+
conv_transpose_2d-CONST_EVAL_PASS-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 2, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
14+
conv_transpose_2d-CONST_EVAL_PASS-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 3, 'output_padding': 0, 'groups': 1, 'bias': True, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None
15+
conv_transpose_2d-FROM_ANOTHER_OP-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
16+
conv_transpose_2d-FROM_ANOTHER_OP-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 2, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
17+
conv_transpose_2d-FROM_ANOTHER_OP-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None
18+
conv_transpose_2d-FROM_ANOTHER_OP-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 3, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None
19+
conv_transpose_2d-FROM_HOST-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
20+
conv_transpose_2d-FROM_HOST-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 2, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
21+
conv_transpose_2d-FROM_HOST-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None
22+
conv_transpose_2d-FROM_HOST-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 3, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None
23+
conv_transpose_2d-CONST_EVAL_PASS-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
24+
conv_transpose_2d-CONST_EVAL_PASS-{'in_channels': 10, 'out_channels': 7, 'kernel_size': (26, 23), 'stride': 1, 'padding': 2, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 10, 1000, 100)-None-None
25+
conv_transpose_2d-CONST_EVAL_PASS-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 0, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None
26+
conv_transpose_2d-CONST_EVAL_PASS-{'in_channels': 64, 'out_channels': 36, 'kernel_size': (26, 9), 'stride': 1, 'padding': 3, 'output_padding': 0, 'groups': 1, 'bias': False, 'dilation': 1, 'dtype': None}-(1, 64, 160, 96)-None-None

0 commit comments

Comments
 (0)