This is an automated email from the ASF dual-hosted git repository. tqchen pushed a commit to branch tvm-further-cleanup-python-tests-followup in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 401960770318c2ea662aa43d37dbad036b10cb87 Author: Tianqi Chen <[email protected]> AuthorDate: Sun Jul 5 19:01:26 2026 +0000 [CI] Partition Python tests by GPU marker --- tests/scripts/task_python_unittest.sh | 2 +- tests/scripts/task_python_unittest_gpuonly.sh | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/scripts/task_python_unittest.sh b/tests/scripts/task_python_unittest.sh index 02d468257c..524a9f2a2e 100755 --- a/tests/scripts/task_python_unittest.sh +++ b/tests/scripts/task_python_unittest.sh @@ -24,4 +24,4 @@ export PYTEST_ADDOPTS="${CI_PYTEST_ADD_OPTIONS:-} ${PYTEST_ADDOPTS:-}" # setup tvm-ffi into python folder uv pip install -v --target=python ./3rdparty/tvm-ffi/ -python3 -m pytest -vvs -n auto tests/python +python3 -m pytest -vvs -n auto -m "${TVM_TEST_MARKER:-not gpu}" tests/python diff --git a/tests/scripts/task_python_unittest_gpuonly.sh b/tests/scripts/task_python_unittest_gpuonly.sh index e6ac397c45..ca1ae86f73 100755 --- a/tests/scripts/task_python_unittest_gpuonly.sh +++ b/tests/scripts/task_python_unittest_gpuonly.sh @@ -19,7 +19,7 @@ set -euxo pipefail # Every GPU test carries the `gpu` marker; the specific backend is gated by skipif. -export PYTEST_ADDOPTS="-m gpu ${PYTEST_ADDOPTS:-}" +export TVM_TEST_MARKER="gpu" # Test most of the enabled runtimes here. # TODO: disabled opencl tests due to segmentation fault. @@ -34,4 +34,5 @@ export TVM_TEST_TARGETS='{"kind":"vulkan","from_device":0}' export PYTHONPATH="$(pwd)/python" export PYTEST_ADDOPTS="${CI_PYTEST_ADD_OPTIONS:-} ${PYTEST_ADDOPTS:-}" -python3 -m pytest -vvs -n auto tests/python/codegen/test_target_codegen_vulkan.py +python3 -m pytest -vvs -n auto -m "${TVM_TEST_MARKER}" \ + tests/python/codegen/test_target_codegen_vulkan.py
