mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
e9bed8b23b
1. Change the python packaing pipeline to use manylinux1 2. Temporarily disable model test in the python pipeline.
52 lines
2.3 KiB
Bash
Executable file
52 lines
2.3 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.
|
|
|
|
# The script build onnx from source because the prebuilt package doesn't conform to manylinux1 standard.
|
|
|
|
set -e
|
|
PYTHON_VER=$1
|
|
|
|
if [[ "$PYTHON_VER" = "3.5" && -d "/opt/python/cp35-cp35m" ]]; then
|
|
PYTHON_EXE="/opt/python/cp35-cp35m/bin/python3.5"
|
|
export PATH=/opt/python/cp35-cp35m/bin:$PATH
|
|
elif [[ "$PYTHON_VER" = "3.6" && -d "/opt/python/cp36-cp36m" ]]; then
|
|
PYTHON_EXE="/opt/python/cp36-cp36m/bin/python3.6"
|
|
export PATH=/opt/python/cp36-cp36m/bin:$PATH
|
|
elif [[ "$PYTHON_VER" = "3.7" && -d "/opt/python/cp37-cp37m" ]]; then
|
|
PYTHON_EXE="/opt/python/cp37-cp37m/bin/python3.7"
|
|
export PATH=/opt/python/cp37-cp37m/bin:$PATH
|
|
else
|
|
PYTHON_EXE="/usr/bin/python${PYTHON_VER}"
|
|
fi
|
|
|
|
version2tag=(5af210ca8a1c73aa6bae8754c9346ec54d0a756e-onnx123
|
|
bae6333e149a59a3faa9c4d9c44974373dcf5256-onnx130
|
|
9e55ace55aad1ada27516038dfbdc66a8a0763db-onnx141
|
|
7d7bc83d29a328233d3e8affa4c4ea8b3e3599ef-onnx150
|
|
553df22c67bee5f0fe6599cff60f1afc6748c635-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_EXE} -m pip uninstall -y onnx
|
|
fi
|
|
lastest_onnx_version=$onnx_version
|
|
GetFile https://github.com/onnx/onnx/archive/$onnx_version.tar.gz /tmp/src/$onnx_version.tar.gz
|
|
tar -xf /tmp/src/$onnx_version.tar.gz -C /tmp/src
|
|
cd /tmp/src/onnx-$onnx_version
|
|
if [ ! -d "third_party/pybind11/pybind11" ]; then
|
|
git clone https://github.com/pybind/pybind11.git third_party/pybind11
|
|
fi
|
|
${PYTHON_EXE} -m pip install .
|
|
mkdir -p /data/onnx/${onnx_tag}
|
|
backend-test-tools generate-data -o /data/onnx/$onnx_tag
|
|
done
|