From 93e8a6fbb56268000b2e0a6704ae0cb12b72c408 Mon Sep 17 00:00:00 2001 From: Arham Khan Date: Wed, 19 Jul 2023 13:21:54 -0500 Subject: [PATCH] build latest torch-mlir in test workflow --- tests/torch_mlir/main.py | 2 +- tests/torch_mlir/torch-mlir-requirements.txt | 2 +- tests/torch_mlir/xfail.py | 1 + 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/torch_mlir/main.py b/tests/torch_mlir/main.py index a086492..3d60932 100644 --- a/tests/torch_mlir/main.py +++ b/tests/torch_mlir/main.py @@ -290,7 +290,7 @@ def test_suite(self, sequential=False): with patch_meta_path(overloads): torch_mlir_register_all_tests() - run_pi_tests(torch_mlir_linalg_module_strs, sequential=False) + run_pi_tests(torch_mlir_linalg_module_strs, sequential=True) if __name__ == "__main__": diff --git a/tests/torch_mlir/torch-mlir-requirements.txt b/tests/torch_mlir/torch-mlir-requirements.txt index 700fac5..c7915e5 100644 --- a/tests/torch_mlir/torch-mlir-requirements.txt +++ b/tests/torch_mlir/torch-mlir-requirements.txt @@ -1,6 +1,6 @@ # Torch-MLIR -f https://llvm.github.io/torch-mlir/package-index/ -torch-mlir==20230616.871 +torch-mlir==20230720.905 # TODO: This is copied from the torch-mlir requirements.txt # Because it requires installing a nightly build of PyTorch, diff --git a/tests/torch_mlir/xfail.py b/tests/torch_mlir/xfail.py index 7fa99d5..2235f22 100644 --- a/tests/torch_mlir/xfail.py +++ b/tests/torch_mlir/xfail.py @@ -14,6 +14,7 @@ "ResNet18StaticModule_basic", "SliceCopyStartGreaterThanDimSize_Module_basic", "UniformModule_basic", + "UniformStaticShapeModule_basic" } PI_XFAIL_SET = {