This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm-tirx.git
The following commit(s) were added to refs/heads/main by this push:
new 52e6142a1 [CI] Run minimal examples in CPU and GPU unit tests (#113)
52e6142a1 is described below
commit 52e6142a1db212ec1bcd2444e85c80a956ba11f7
Author: Tianqi Chen <[email protected]>
AuthorDate: Wed Oct 28 13:05:04 2026 -0400
[CI] Run minimal examples in CPU and GPU unit tests (#113)
---
ci/jenkins/cpu_jenkinsfile.groovy | 1 +
ci/jenkins/gpu_jenkinsfile.groovy | 1 +
tests/scripts/task_minimal_examples.sh | 30 ++++++++++++++++++++++++++++++
3 files changed, 32 insertions(+)
diff --git a/ci/jenkins/cpu_jenkinsfile.groovy
b/ci/jenkins/cpu_jenkinsfile.groovy
index 3f143705b..8f91c9f32 100644
--- a/ci/jenkins/cpu_jenkinsfile.groovy
+++ b/ci/jenkins/cpu_jenkinsfile.groovy
@@ -57,6 +57,7 @@ stage('Test') {
script: """${docker_run} ${images.ci_cpu} bash -euo pipefail -c '
source ./tests/scripts/test_task_install_dist_wheels.sh
./tests/scripts/task_python_unittest.sh
+ ./tests/scripts/task_minimal_examples.sh cpu
'""",
label: 'Run Python unit tests',
)
diff --git a/ci/jenkins/gpu_jenkinsfile.groovy
b/ci/jenkins/gpu_jenkinsfile.groovy
index 523447ee7..9f2374259 100644
--- a/ci/jenkins/gpu_jenkinsfile.groovy
+++ b/ci/jenkins/gpu_jenkinsfile.groovy
@@ -66,6 +66,7 @@ parallel(
script: """${docker_run} ${images.ci_gpu} bash -euo pipefail -c '
source ./tests/scripts/test_task_install_dist_wheels.sh --group
cuda
./tests/scripts/task_python_unittest_gpuonly.sh
+ ./tests/scripts/task_minimal_examples.sh cuda
'""",
label: 'Run Python GPU unit tests',
)
diff --git a/tests/scripts/task_minimal_examples.sh
b/tests/scripts/task_minimal_examples.sh
new file mode 100755
index 000000000..481d4c774
--- /dev/null
+++ b/tests/scripts/task_minimal_examples.sh
@@ -0,0 +1,30 @@
+#!/usr/bin/env bash
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+set -euxo pipefail
+
+if [[ $# -ne 1 || ( "$1" != cpu && "$1" != cuda ) ]]; then
+ echo "Usage: $0 cpu|cuda" >&2
+ exit 1
+fi
+
+# Keep this explicit list limited to examples that run in a few seconds.
+examples=(examples/quick_start)
+for example in "${examples[@]}"; do
+ uv run --no-sync bash "${example}/run_all_$1.sh"
+done
