From b9674699066001115252e078942f8239c44a2458 Mon Sep 17 00:00:00 2001 From: Yuanqiang Liu Date: Wed, 15 Mar 2023 00:32:44 +0800 Subject: [PATCH] [e2e] fix stack e2e test typo (#1931) --- python/torch_mlir_e2e_test/test_suite/basic.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/torch_mlir_e2e_test/test_suite/basic.py b/python/torch_mlir_e2e_test/test_suite/basic.py index 96b67e09c..e8c4d1ca9 100644 --- a/python/torch_mlir_e2e_test/test_suite/basic.py +++ b/python/torch_mlir_e2e_test/test_suite/basic.py @@ -660,7 +660,7 @@ class TensorsStackModule(torch.nn.Module): ([-1, -1, -1], torch.float32, True), ]) def forward(self, x, y, z): - return torch.stack([x, y, z], 1) + return torch.stack([x, y, z], dim=1) @register_test_case(module_factory=lambda: TensorsStackModule()) @@ -708,7 +708,7 @@ class TensorsStackPromoteDTypeModule(torch.nn.Module): ([-1, -1, -1], torch.int64, True), ]) def forward(self, x, y, z): - return torch.cat([x, y, z], dim=-2) + return torch.stack([x, y, z], dim=-2) @register_test_case(module_factory=lambda: TensorsStackPromoteDTypeModule())