[tvm] branch main updated: [CI][AArch64] Skip libgomp failures in integration tests (#12554)

[mousius]

This is an automated email from the ASF dual-hosted git repository.

mousius pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new d271678388 [CI][AArch64] Skip libgomp failures in integration tests 
(#12554)
d271678388 is described below

commit d27167838888ca79eb53cd16449ae4483c9b6249
Author: Leandro Nunes <leandro.nu...@arm.com>
AuthorDate: Tue Aug 23 14:03:04 2022 +0100

    [CI][AArch64] Skip libgomp failures in integration tests (#12554)
    
    Some integration tests are failing when running in CI machines that
    have torch installed (validated only in AARch64 for now), with an
    error message related to libgomp, similar to the one above:
    
    OSError: /.../dist-packages/torch/lib/libgomp-d22c30c5.so.1: cannot
    allocate memory in static TLS block
    
    As part of enabling the integration tests in AArch64, I'm marking this
    tests as skipped, so that tests can start executing and don't regress
    while we take time to investigate these specific failures.
---
 tests/python/driver/tvmc/test_autotuner.py | 9 +++++++++
 tests/python/driver/tvmc/test_frontends.py | 9 +++++++++
 tests/python/driver/tvmc/test_model.py     | 4 ++++
 3 files changed, 22 insertions(+)

diff --git a/tests/python/driver/tvmc/test_autotuner.py 
b/tests/python/driver/tvmc/test_autotuner.py
index 66017823a6..7c05ff804f 100644
--- a/tests/python/driver/tvmc/test_autotuner.py
+++ b/tests/python/driver/tvmc/test_autotuner.py
@@ -14,6 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+import platform
 import pytest
 import os
 
@@ -73,6 +74,10 @@ def test_get_tuning_tasks(onnx_mnist):
     assert all([type(x) is expected_task_type for x in sut]) is True
 
 
+@pytest.mark.skipif(
+    platform.machine() == "aarch64",
+    reason="Currently failing on AArch64 - see 
https://github.com/apache/tvm/issues/10673";,
+)
 def test_tune_tasks__tuner__xgb(onnx_mnist, tmpdir_factory):
     pytest.importorskip("onnx")
 
@@ -141,6 +146,10 @@ def 
test_tune_tasks__tuner__xgb__no_early_stopping(onnx_mnist, tmpdir_factory):
     _tuner_test_helper(onnx_mnist, "xgb", tmpdir_name, early_stopping=None)
 
 
+@pytest.mark.skipif(
+    platform.machine() == "aarch64",
+    reason="Currently failing on AArch64 - see 
https://github.com/apache/tvm/issues/10673";,
+)
 def test_tune_tasks__tuner__xgb__no_tuning_records(onnx_mnist, tmpdir_factory):
     pytest.importorskip("onnx")
 
diff --git a/tests/python/driver/tvmc/test_frontends.py 
b/tests/python/driver/tvmc/test_frontends.py
index 0cd02181ac..98659b05ae 100644
--- a/tests/python/driver/tvmc/test_frontends.py
+++ b/tests/python/driver/tvmc/test_frontends.py
@@ -15,6 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 
+import platform
 import pytest
 import builtins
 import importlib
@@ -74,6 +75,10 @@ def test_guess_frontend_onnx():
     assert type(sut) is tvmc.frontends.OnnxFrontend
 
 
+@pytest.mark.skipif(
+    platform.machine() == "aarch64",
+    reason="Currently failing on AArch64 - see 
https://github.com/apache/tvm/issues/10673";,
+)
 def test_guess_frontend_pytorch():
     # some CI environments wont offer pytorch, so skip in case it is not 
present
     pytest.importorskip("torch")
@@ -245,6 +250,10 @@ def test_load_model__pth(pytorch_resnet18):
     assert "layer1.0.conv1.weight" in tvmc_model.params.keys()
 
 
+@pytest.mark.skipif(
+    platform.machine() == "aarch64",
+    reason="Currently failing on AArch64 - see 
https://github.com/apache/tvm/issues/10673";,
+)
 def test_load_quantized_model__pth(pytorch_mobilenetv2_quantized):
     # some CI environments wont offer torch, so skip in case it is not present
     pytest.importorskip("torch")
diff --git a/tests/python/driver/tvmc/test_model.py 
b/tests/python/driver/tvmc/test_model.py
index 74c1c4ded8..fb1f718c1b 100644
--- a/tests/python/driver/tvmc/test_model.py
+++ b/tests/python/driver/tvmc/test_model.py
@@ -55,6 +55,10 @@ def test_tvmc_workflow(use_vm, keras_simple):
     assert "output_0" in result.outputs.keys()
 
 
+@pytest.mark.skipif(
+    platform.machine() == "aarch64",
+    reason="Currently failing on AArch64 - see 
https://github.com/apache/tvm/issues/10673";,
+)
 @pytest.mark.parametrize("use_vm", [True, False])
 def test_save_load_model(use_vm, keras_simple, tmpdir_factory):
     pytest.importorskip("onnx")