mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-15 20:50:42 +00:00
5e1244eb4d
* Update to ONNX submodule to ONNX 1.7 release branch. * Update to ONNX submodule to ONNX 1.7 release branch. * fix version.
62 lines
3 KiB
Bash
Executable file
62 lines
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"
|
|
elif [[ "$PYTHON_VER" = "3.6" && -d "/opt/python/cp36-cp36m" ]]; then
|
|
PYTHON_EXE="/opt/python/cp36-cp36m/bin/python3.6"
|
|
elif [[ "$PYTHON_VER" = "3.7" && -d "/opt/python/cp37-cp37m" ]]; then
|
|
PYTHON_EXE="/opt/python/cp37-cp37m/bin/python3.7"
|
|
elif [[ "$PYTHON_VER" = "3.8" && -d "/opt/python/cp38-cp38" ]]; then
|
|
PYTHON_EXE="/opt/python/cp38-cp38/bin/python3.8"
|
|
else
|
|
PYTHON_EXE="/usr/bin/python${PYTHON_VER}"
|
|
fi
|
|
|
|
${PYTHON_EXE} -m pip install protobuf
|
|
|
|
version2tag=(5af210ca8a1c73aa6bae8754c9346ec54d0a756e-onnx123
|
|
bae6333e149a59a3faa9c4d9c44974373dcf5256-onnx130
|
|
9e55ace55aad1ada27516038dfbdc66a8a0763db-onnx141
|
|
7d7bc83d29a328233d3e8affa4c4ea8b3e3599ef-onnx150
|
|
1facb4c1bb9cc2107d4dbaf9fd647fefdbbeb0ab-onnx161
|
|
0c070abb0c40fec649f81a73a75b0098662ec486-onnx170) #1.7.0
|
|
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
|
|
# We need to make the adjustment only for CentOS6 OR we substitue this only for
|
|
# ${PYTHON_EXE} where we'd need to escape slashes
|
|
# Make sure we do not hit pyhon2 as on CentOS 6 it does not work
|
|
ESCAPED_PY=$(echo "${PYTHON_EXE}" | sed 's/\//\\\//g')
|
|
sed "1,1 s/\/usr\/bin\/env python/$ESCAPED_PY/" /tmp/src/onnx-$onnx_version/tools/protoc-gen-mypy.py > /tmp/src/onnx-$onnx_version/tools/repl_protoc-gen-mypy.py
|
|
chmod a+w /tmp/src/onnx-$onnx_version/tools/protoc-gen-mypy.py
|
|
mv /tmp/src/onnx-$onnx_version/tools/repl_protoc-gen-mypy.py /tmp/src/onnx-$onnx_version/tools/protoc-gen-mypy.py
|
|
mkdir -p /data/onnx/${onnx_tag}
|
|
${PYTHON_EXE} -m pip install .
|
|
cd /tmp
|
|
${PYTHON_EXE} -m onnx.backend.test.cmd_tools generate-data -o /data/onnx/$onnx_tag
|
|
done
|