From 13b02d8c9506654f8530fd8f43d139e719253e5e Mon Sep 17 00:00:00 2001 From: Leandro Nunes Date: Thu, 8 Sep 2022 09:32:57 +0100 Subject: [PATCH] [CI][AArch64] Mark tests to be skipped due to torch crash Some integration tests are not being run on CI due to the configuration of the machine with onnx and torch not calling the integration tests script. This patch skips two more tests failing with the error message below: ``` "OSError: /.../torch/lib/libgomp-d22c30c5.so.1: cannot allocate memory in static TLS block" ``` Change-Id: I9e217e7e2733836e408ec86a0ca828d23d5752d2 --- tests/python/driver/tvmc/test_frontends.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/python/driver/tvmc/test_frontends.py b/tests/python/driver/tvmc/test_frontends.py index 1ccac7696fcc..c1a3be67c208 100644 --- a/tests/python/driver/tvmc/test_frontends.py +++ b/tests/python/driver/tvmc/test_frontends.py @@ -237,6 +237,10 @@ def test_load_model___wrong_language__to_onnx(tflite_mobilenet_v1_1_quant): tvmc.load(tflite_mobilenet_v1_1_quant, model_format="onnx") +@pytest.mark.skipif( + platform.machine() == "aarch64", + reason="Currently failing on AArch64 - see https://github.com/apache/tvm/issues/10673", +) def test_load_model__pth(pytorch_resnet18): # some CI environments wont offer torch, so skip in case it is not present pytest.importorskip("torch") @@ -432,6 +436,10 @@ def test_import_tensorflow_friendly_message(pb_mobilenet_v1_1_quant, monkeypatch _ = tvmc.frontends.load_model(pb_mobilenet_v1_1_quant, model_format="pb") +@pytest.mark.skipif( + platform.machine() == "aarch64", + reason="Currently failing on AArch64 - see https://github.com/apache/tvm/issues/10673", +) def test_import_torch_friendly_message(pytorch_resnet18, monkeypatch): monkeypatch.setattr("importlib.import_module", mock_error_on_name("torch"))