Skip to content

Commit 95d4bbf

Browse files
committed
remove unflatten tests
1 parent d83ae9a commit 95d4bbf

File tree

2 files changed

+0
-114
lines changed

2 files changed

+0
-114
lines changed

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

Lines changed: 0 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -1281,46 +1281,6 @@ def UnflattenIntNegativeOneSizeStaticModule_basic(module, tu: TestUtils):
12811281
module.forward(tu.rand(5, 12, 3))
12821282

12831283

1284-
class UnflattenIntDynamicModule(torch.nn.Module):
1285-
def __init__(self):
1286-
super().__init__()
1287-
1288-
@export
1289-
@annotate_args(
1290-
[
1291-
None,
1292-
([-1, 12], torch.float32, True),
1293-
]
1294-
)
1295-
def forward(self, inputs):
1296-
return torch.ops.aten.unflatten(inputs, 1, [3, 4])
1297-
1298-
1299-
@register_test_case(module_factory=lambda: UnflattenIntDynamicModule())
1300-
def UnflattenIntDynamicModule_basic(module, tu: TestUtils):
1301-
module.forward(tu.rand(2, 12))
1302-
1303-
1304-
class UnflattenIntDynamicWithInferredSizeModule(torch.nn.Module):
1305-
def __init__(self):
1306-
super().__init__()
1307-
1308-
@export
1309-
@annotate_args(
1310-
[
1311-
None,
1312-
([-1, 20], torch.float32, True),
1313-
]
1314-
)
1315-
def forward(self, inputs):
1316-
return torch.ops.aten.unflatten(inputs, 1, [4, -1])
1317-
1318-
1319-
@register_test_case(module_factory=lambda: UnflattenIntDynamicWithInferredSizeModule())
1320-
def UnflattenIntDynamicWithInferredSizeModule_basic(module, tu: TestUtils):
1321-
module.forward(tu.rand(3, 20))
1322-
1323-
13241284
# ==============================================================================
13251285

13261286

test/Conversion/TorchToLinalg/unflatten.mlir

Lines changed: 0 additions & 74 deletions
This file was deleted.

0 commit comments

Comments
 (0)