mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-15 20:50:42 +00:00
34 lines
1.6 KiB
Bash
Executable file
34 lines
1.6 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
# The script is to generate all supported versions of onnx models which will be tested by onnx_test_runner
|
|
# in the end of ci build pipeline. The purpose is to make sure latest onnxruntime has no regressions. Note
|
|
# that the order of installation must be onnx123, onnx130, onnx141, onnx150 and onnxtip since we want
|
|
# to keep the tip of master on script exit for onnx backend test which is also a part of build pipeline.
|
|
# One possible improvement here is to keep the models saved to some public storage instead of generating
|
|
# on the fly every time.
|
|
|
|
set -e
|
|
PYTHON_VER=$1
|
|
version2tag=(5af210ca8a1c73aa6bae8754c9346ec54d0a756e-onnx123
|
|
bae6333e149a59a3faa9c4d9c44974373dcf5256-onnx130
|
|
9e55ace55aad1ada27516038dfbdc66a8a0763db-onnx141
|
|
7d7bc83d29a328233d3e8affa4c4ea8b3e3599ef-onnx150
|
|
2873fea8bd4679fdf337d64d1140e80a61986c29-onnxtip)
|
|
for v2t in ${version2tag[*]}; do
|
|
onnx_version="$(cut -d'-' -f1<<<${v2t})"
|
|
onnx_tag="$(cut -d'-' -f2<<<${v2t})"
|
|
if [ -z ${lastest_onnx_version+x} ]; then
|
|
echo "first pass";
|
|
else
|
|
echo "deleting old onnx-${lastest_onnx_version}";
|
|
python${PYTHON_VER} -m pip uninstall -y onnx
|
|
fi
|
|
lastest_onnx_version=$onnx_version
|
|
aria2c -q -d /tmp/src https://github.com/onnx/onnx/archive/$onnx_version.tar.gz
|
|
tar -xf /tmp/src/onnx-$onnx_version.tar.gz -C /tmp/src
|
|
cd /tmp/src/onnx-$onnx_version
|
|
git clone https://github.com/pybind/pybind11.git third_party/pybind11
|
|
python${PYTHON_VER} -m pip install .
|
|
mkdir -p /data/onnx/${onnx_tag}
|
|
backend-test-tools generate-data -o /data/onnx/$onnx_tag
|
|
done
|