Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion ci/jenkins/docker-images.ini
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@

# This data file is read during when Jenkins runs job to determine docker images.
[jenkins]
ci_tag: 20260219-160550-72f51851
ci_tag: 20260301-134651-63f099ad
ci_arm: tlcpack/ci-arm:%(ci_tag)s
ci_cpu: tlcpack/ci_cpu:%(ci_tag)s
ci_gpu: tlcpack/ci-gpu:%(ci_tag)s
Expand Down
3 changes: 3 additions & 0 deletions tests/python/relax/test_frontend_onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -3016,6 +3016,9 @@ def test_onehot():
@pytest.mark.parametrize("sorted", [0, 1])
@pytest.mark.parametrize("num_outputs", [1, 2, 3, 4])
def test_unique(axis: int | None, sorted: int, num_outputs: int):
if num_outputs in [3, 4] and axis is None:
pytest.xfail("RuntimeError: Check failed: input_shape.size() == size (2 vs. 1)")

input_shape = [8, 8]
if axis is None:
output_shape = [-1]
Expand Down
59 changes: 1 addition & 58 deletions tests/scripts/ci.py
Original file line number Diff line number Diff line change
Expand Up @@ -285,7 +285,7 @@ def docs(
]

extra_setup = [
"python3 -m pip install " + " ".join(requirements),
"uv pip install " + " ".join(requirements),
]
else:
check_gpu()
Expand Down Expand Up @@ -594,7 +594,6 @@ def add_subparser(
"./tests/scripts/task_python_integration_gpuonly.sh",
],
),
"frontend": ("run frontend tests", ["./tests/scripts/task_python_frontend.sh"]),
},
),
generate_command(
Expand All @@ -612,20 +611,6 @@ def add_subparser(
"./tests/scripts/task_python_unittest.sh",
],
),
"frontend": ("run frontend tests", ["./tests/scripts/task_python_frontend_cpu.sh"]),
},
),
generate_command(
name="minimal",
help="Run minimal CPU build and test(s)",
options={
"cpp": CPP_UNITTEST,
"unittest": (
"run unit tests",
[
"./tests/scripts/task_python_unittest.sh",
],
),
},
),
generate_command(
Expand All @@ -650,48 +635,6 @@ def add_subparser(
),
},
),
generate_command(
name="adreno",
help="Run Adreno build and test(s)",
post_build=["./tests/scripts/task_build_adreno_bins.sh"],
additional_flags={
"--volume": os.environ.get("ADRENO_OPENCL", "/tmp/") + ":/adreno-opencl",
"--net": "host",
},
env={
"ADRENO_OPENCL": "/adreno-opencl",
"ADRENO_TARGET_CLML_VERSION": os.environ.get("ADRENO_TARGET_CLML_VERSION", "3"),
},
options={
"test": (
"run Adreno API/Python tests",
[
"./tests/scripts/task_python_adreno.sh " + os.environ.get("ANDROID_SERIAL", ""),
],
),
"benchmarks": (
"run Adreno Benchmarks (Native OpenCL, CLML SDK)",
[
"./apps/benchmark/adreno/bench.sh texture "
+ os.environ.get("ANDROID_SERIAL", ""),
"./apps/benchmark/adreno/bench.sh clml " + os.environ.get("ANDROID_SERIAL", ""),
],
),
"nativebenchmarks": (
"run Adreno Texture Benchmarks",
[
"./apps/benchmark/adreno/bench.sh texture "
+ os.environ.get("ANDROID_SERIAL", ""),
],
),
"clmlbenchmarks": (
"run Adreno CLML SDK Benchmarks",
[
"./apps/benchmark/adreno/bench.sh clml " + os.environ.get("ANDROID_SERIAL", ""),
],
),
},
),
]


Expand Down
2 changes: 0 additions & 2 deletions tests/scripts/task_config_build_arm.sh
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,5 @@ echo set\(USE_SORT ON\) >> config.cmake
echo set\(USE_RPC ON\) >> config.cmake
echo set\(USE_LLVM llvm-config-17\) >> config.cmake
echo set\(CMAKE_CXX_FLAGS -Werror\) >> config.cmake
echo set\(USE_ARM_COMPUTE_LIB ON\) >> config.cmake
echo set\(USE_ARM_COMPUTE_LIB_GRAPH_EXECUTOR "/opt/acl"\) >> config.cmake
echo set\(USE_CCACHE OFF\) >> config.cmake
echo set\(SUMMARIZE ON\) >> config.cmake
12 changes: 1 addition & 11 deletions tests/scripts/task_config_build_cpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -25,21 +25,11 @@ cp ../cmake/config.cmake .

echo set\(USE_SORT ON\) >> config.cmake
echo set\(USE_DNNL ON\) >> config.cmake
echo set\(USE_ARM_COMPUTE_LIB ON\) >> config.cmake
echo set\(USE_LLVM \"/usr/bin/llvm-config-17 --link-static\"\) >> config.cmake
echo set\(CMAKE_CXX_FLAGS \"-Wno-error=range-loop-construct -Wno-error=comment\"\) >> config.cmake
echo set\(HIDE_PRIVATE_SYMBOLS ON\) >> config.cmake

# This conditional is just to support the transition to cope
# with the change in the way TFLite is built. It can be
# removed once we migrate to TensorFlow and TFLite > 2.9.1
if [ -d "/opt/tflite" ]; then
echo set\(USE_TFLITE \"/opt/tflite\"\) >> config.cmake
else
echo set\(USE_TFLITE ON\) >> config.cmake
fi

echo set\(USE_TENSORFLOW_PATH \"/tensorflow\"\) >> config.cmake
echo set\(USE_TFLITE \"/opt/tflite\"\) >> config.cmake
echo set\(USE_FLATBUFFERS_PATH \"/flatbuffers\"\) >> config.cmake
echo set\(USE_CCACHE OFF\) >> config.cmake
echo set\(SUMMARIZE ON\) >> config.cmake
4 changes: 2 additions & 2 deletions tests/scripts/task_python_docs.sh
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ sphinx_precheck() {
echo "PreCheck sphinx doc generation WARNINGS.."

# setup tvm-ffi into python folder
python3 -m pip install -v --target=python ./3rdparty/tvm-ffi/
uv pip install -v --target=python ./3rdparty/tvm-ffi/

pushd docs
make clean
Expand Down Expand Up @@ -127,7 +127,7 @@ find . -type f -path "*.log" | xargs rm -f
find . -type f -path "*.pyc" | xargs rm -f

# setup tvm-ffi into python folder
python3 -m pip install -v --target=python ./3rdparty/tvm-ffi/
uv pip install -v --target=python ./3rdparty/tvm-ffi/


cd docs
Expand Down
2 changes: 1 addition & 1 deletion tests/scripts/task_python_integration.sh
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ fi
find . -type f -path "*.pyc" | xargs rm -f

# setup tvm-ffi into python folder
python3 -m pip install -v --target=python ./3rdparty/tvm-ffi/
uv pip install -v --target=python ./3rdparty/tvm-ffi/

# Test for OpenCLML
pytest tests/python/relax/backend/clml/
2 changes: 1 addition & 1 deletion tests/scripts/task_python_nightly.sh
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ set -euxo pipefail
source tests/scripts/setup-pytest-env.sh

# setup tvm-ffi into python folder
python3 -m pip install -v --target=python ./3rdparty/tvm-ffi/
uv pip install -v --target=python ./3rdparty/tvm-ffi/

# cleanup pycache
find . -type f -path "*.pyc" | xargs rm -f
Expand Down
2 changes: 1 addition & 1 deletion tests/scripts/task_python_unittest.sh
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ source tests/scripts/setup-pytest-env.sh
find . -type f -path "*.pyc" | xargs rm -f

# setup tvm-ffi into python folder
python3 -m pip install -v --target=python ./3rdparty/tvm-ffi/
uv pip install -v --target=python ./3rdparty/tvm-ffi/

# NOTE: also set by task_python_unittest_gpuonly.sh.
if [ -z "${TVM_UNITTEST_TESTSUITE_NAME:-}" ]; then
Expand Down
2 changes: 1 addition & 1 deletion tests/scripts/task_web_wasm.sh
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ set -euxo pipefail
export PYTHONPATH=`pwd`/python

# setup tvm-ffi into python folder
python3 -m pip install -v --target=python ./3rdparty/tvm-ffi/
uv pip install -v --target=python ./3rdparty/tvm-ffi/

rm -rf .emscripten_cache
cd web
Expand Down
Loading