From 8d9173a006c415a132e672bacabcb7bc471acdc0 Mon Sep 17 00:00:00 2001 From: Qingchao Shen Date: Mon, 22 May 2023 13:10:01 +0800 Subject: [PATCH 1/2] fix test cases MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Add some version check to avoid calling the operator/attribute which has not been introduced yet！ --- tests/python/frontend/pytorch/test_forward.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py index 7552f5cc61d2..1ce5aae2e821 100644 --- a/tests/python/frontend/pytorch/test_forward.py +++ b/tests/python/frontend/pytorch/test_forward.py @@ -1012,11 +1012,13 @@ def __init__(self, split_size_or_sections, dim): def forward(self, *args): return torch.tensor_split(args[0], self.split_size_or_sections, self.dim) - input_data = torch.rand(input_shape).float() - verify_model(Tensor_Split(2, 0).float().eval(), input_data=input_data) - verify_model(Tensor_Split(torch.tensor(3), 1).float().eval(), input_data=input_data) - verify_model(Tensor_Split([2, 3, 5], 1).float().eval(), input_data=input_data) - verify_model(Tensor_Split((2, 3, 5), 1).float().eval(), input_data=input_data) + # tensor_split was introduced when torch > 1.7.1 + if package_version.parse(torch.__version__) > package_version.parse("1.7.1"): + input_data = torch.rand(input_shape).float() + verify_model(Tensor_Split(2, 0).float().eval(), input_data=input_data) + verify_model(Tensor_Split(torch.tensor(3), 1).float().eval(), input_data=input_data) + verify_model(Tensor_Split([2, 3, 5], 1).float().eval(), input_data=input_data) + verify_model(Tensor_Split((2, 3, 5), 1).float().eval(), input_data=input_data) @tvm.testing.uses_gpu @@ -5025,6 +5027,10 @@ def forward(self, x, y): grid_3D = torch.rand([4, 8, 8, 8, 3]).float() for _method in methods: + # bicubic was introduced when pytorch > 1.7.1 + torch_version = package_version.parse(torch.__version__) + if _method=='bicubic' and torch_version <= package_version.parse("1.7.1"): + continue for _padding in padding_modes: for _align in align_corners: # ATTENTION: From 2d71701e731c360f681a4298ee1e96b757b5832c Mon Sep 17 00:00:00 2001 From: Qingchao Shen Date: Mon, 22 May 2023 14:42:36 +0800 Subject: [PATCH 2/2] Update test_forward.py --- tests/python/frontend/pytorch/test_forward.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py index 1ce5aae2e821..b602c14df3af 100644 --- a/tests/python/frontend/pytorch/test_forward.py +++ b/tests/python/frontend/pytorch/test_forward.py @@ -5029,7 +5029,7 @@ def forward(self, x, y): for _method in methods: # bicubic was introduced when pytorch > 1.7.1 torch_version = package_version.parse(torch.__version__) - if _method=='bicubic' and torch_version <= package_version.parse("1.7.1"): + if _method == "bicubic" and torch_version <= package_version.parse("1.7.1"): continue for _padding in padding_modes: for _align in align_corners: