Skip to content

Commit

Permalink
build latest torch-mlir in test workflow
Browse files Browse the repository at this point in the history
  • Loading branch information
Arham Khan authored and Arham Khan committed Jul 20, 2023
1 parent b1becd0 commit 93e8a6f
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 2 deletions.
2 changes: 1 addition & 1 deletion tests/torch_mlir/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__":
Expand Down
2 changes: 1 addition & 1 deletion tests/torch_mlir/torch-mlir-requirements.txt
Original file line number Diff line number Diff line change
@@ -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,
Expand Down
1 change: 1 addition & 0 deletions tests/torch_mlir/xfail.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
"ResNet18StaticModule_basic",
"SliceCopyStartGreaterThanDimSize_Module_basic",
"UniformModule_basic",
"UniformStaticShapeModule_basic"
}

PI_XFAIL_SET = {
Expand Down

0 comments on commit 93e8a6f

Please sign in to comment.