Skip to content

Commit 3b8a12d

Browse files
author
Hariprasad Ravishankar
committed
Added additional e2e test for transposedConv1d
1 parent 3616eef commit 3b8a12d

File tree

1 file changed

+32
-0
lines changed
  • projects/pt1/python/torch_mlir_e2e_test/test_suite

1 file changed

+32
-0
lines changed

projects/pt1/python/torch_mlir_e2e_test/test_suite/conv.py

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2002,6 +2002,38 @@ def TransposedConv1dNegativePadding_basic(module, tu: TestUtils):
20022002
module.forward(tu.rand(1, 1, 7), tu.rand(1, 2, 3), tu.rand(2))
20032003

20042004

2005+
class ConvTranspose1dNegativePaddingLarge(torch.nn.Module):
2006+
def __init__(self):
2007+
super().__init__()
2008+
2009+
@export
2010+
@annotate_args(
2011+
[
2012+
None,
2013+
([1, 17, 5], torch.float32, True),
2014+
([17, 6, 3], torch.float32, True),
2015+
([6], torch.float32, True),
2016+
]
2017+
)
2018+
def forward(self, inputVec, weight, bias):
2019+
return torch.ops.aten.convolution(
2020+
inputVec,
2021+
weight,
2022+
bias=bias,
2023+
stride=[7],
2024+
padding=[10],
2025+
dilation=[4],
2026+
transposed=True,
2027+
output_padding=[0],
2028+
groups=1,
2029+
)
2030+
2031+
2032+
@register_test_case(module_factory=lambda: ConvTranspose1dNegativePaddingLarge())
2033+
def ConvTranspose1dNegativePaddingLarge_basic(module, tu: TestUtils):
2034+
module.forward(tu.rand(1, 17, 5), tu.rand(17, 6, 3), tu.rand(6))
2035+
2036+
20052037
class TransposedConv2dNegativePadding(torch.nn.Module):
20062038
def __init__(self):
20072039
super().__init__()

0 commit comments

Comments
 (0)