diff --git a/Jenkinsfile b/Jenkinsfile index 65ccbf27326f..8fd8faaaf721 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -46,7 +46,7 @@ // NOTE: these lines are scanned by docker/dev_common.sh. Please update the regex as needed. --> ci_lint = "tlcpack/ci-lint:v0.66" ci_gpu = "tlcpack/ci-gpu:v0.76" -ci_cpu = "tlcpack/ci-cpu:v0.75" +ci_cpu = "mbrookhart/ci-cpu:v0.3" ci_wasm = "tlcpack/ci-wasm:v0.71" ci_i386 = "tlcpack/ci-i386:v0.73" ci_qemu = "tlcpack/ci-qemu:v0.06" diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py index 8b633c18977a..b3f9a8225456 100644 --- a/tests/python/frontend/onnx/test_forward.py +++ b/tests/python/frontend/onnx/test_forward.py @@ -4693,6 +4693,8 @@ def verify_eyelike(indata): "test_unsqueeze_two_axes/", "test_unsqueeze_unsorted_axes/", "test_upsample_nearest/", + "test_range_float_type_positive_delta_expanded/", + "test_range_int32_type_negative_delta_expanded", ] @@ -4703,12 +4705,15 @@ def verify_eyelike(indata): "test_mod_mixed_sign_float16/", "test_qlinearconv/", "test_resize_upsample_sizes_nearest/", + "test_basic_convinteger/", + "test_convinteger_with_padding/", ] } @pytest.mark.parametrize("target", targets) @pytest.mark.parametrize("test", onnx_test_folders) +@tvm.testing.uses_gpu def test_onnx_nodes(test, target): if target in target_skips: for failure in target_skips[target]: diff --git a/tests/scripts/task_python_frontend.sh b/tests/scripts/task_python_frontend.sh index fb388a6b7edd..28989b1086df 100755 --- a/tests/scripts/task_python_frontend.sh +++ b/tests/scripts/task_python_frontend.sh @@ -34,10 +34,10 @@ make cython3 echo "Running relay MXNet frontend test..." run_pytest cython python-frontend-mxnet tests/python/frontend/mxnet -echo "Running relay ONNX frontend test..." # Enable tvm.testing decorators in the ONNX importer test (not enabling in the other tests because we # they do not consistently use the decorators to indicate that tests should run on GPU) # In the future, we should enable tvm.testing decorators for all the test files. +echo "Running relay ONNX frontend test (on GPU only)..." PYTEST_ADDOPTS="-m gpu $PYTEST_ADDOPTS" run_pytest cython python-frontend-onnx tests/python/frontend/onnx echo "Running relay CoreML frontend test..." diff --git a/tests/scripts/task_python_frontend_cpu.sh b/tests/scripts/task_python_frontend_cpu.sh index 208714c64988..baabd67c3739 100755 --- a/tests/scripts/task_python_frontend_cpu.sh +++ b/tests/scripts/task_python_frontend_cpu.sh @@ -32,6 +32,9 @@ find . -type f -path "*.pyc" | xargs rm -f # Rebuild cython make cython3 +echo "Running relay ONNX frontend test..." +run_pytest cython python-frontend-onnx tests/python/frontend/onnx + echo "Running relay TFLite frontend test..." run_pytest cython python-frontend-tflite tests/python/frontend/tflite