saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-27 03:11:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Python backend: use packaging.version to parse ONNX version (#11800)

Browse source 
Unlike the previous code, this handles version strings like "1.12.0rc3".

Unblocks https://github.com/microsoft/onnxruntime/issues/11640.
This commit is contained in:
Gary Miguel 2022-06-14 10:17:35 -07:00 committed by GitHub
parent f6d2b629a0
commit 52f6db19da
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
onnxruntime/python/backend/backend.py
View file

@ -8,6 +8,7 @@ Implements ONNX's backend API.
import os
import unittest
import packaging.version
from onnx import ModelProto, helper, version
from onnx.backend.base import Backend
from onnx.checker import check_model
@ -127,8 +128,8 @@ class OnnxRuntimeBackend(Backend):
# check_model serializes the model anyways, so serialize the model once here
# and reuse it below in the cls.prepare call to avoid an additional serialization
# only works with onnx >= 1.10.0 hence the version check
onnx_version = tuple(map(int, (version.version.split(".")[:3])))
onnx_supports_serialized_model_check = onnx_version >= (1, 10, 0)
onnx_version = packaging.version.parse(version.version) or packaging.version.Version("0")
onnx_supports_serialized_model_check = onnx_version.release >= (1, 10, 0)
bin_or_model = model.SerializeToString() if onnx_supports_serialized_model_check else model
check_model(bin_or_model)
opset_supported, error_message = cls.is_opset_supported(model)