Skip to content
This repository has been archived by the owner on Nov 25, 2022. It is now read-only.

Commit

Permalink
[CI][AArch64] Skip libgomp failures in integration tests (apache#12554)
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
leandron authored and xinetzone committed Nov 25, 2022
1 parent 9974903 commit 3ba4e69
Show file tree
Hide file tree
Showing 3 changed files with 22 additions and 0 deletions.
9 changes: 9 additions & 0 deletions tests/python/driver/tvmc/test_autotuner.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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")

Expand Down Expand Up @@ -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")

Expand Down
9 changes: 9 additions & 0 deletions tests/python/driver/tvmc/test_frontends.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.

import platform
import pytest
import builtins
import importlib
Expand Down Expand Up @@ -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")
Expand Down Expand Up @@ -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")
Expand Down
4 changes: 4 additions & 0 deletions tests/python/driver/tvmc/test_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down

0 comments on commit 3ba4e69

Please sign in to comment.