Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
84ce66ec
Unverified
Commit
84ce66ec
authored
Jan 10, 2023
by
Charlie Lin
Committed by
GitHub
Jan 10, 2023
Browse files
Update python ONNX version (#1501)
Change to python ONNX package 1.10.0 to handle new ONNX test generation.
parent
ad1cfd92
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
14 additions
and
1 deletion
+14
-1
test/py/onnx_backend_test.py
test/py/onnx_backend_test.py
+13
-0
tools/install_prereqs.sh
tools/install_prereqs.sh
+1
-1
No files found.
test/py/onnx_backend_test.py
View file @
84ce66ec
...
@@ -94,6 +94,16 @@ def disabled_tests_onnx_1_8_1(backend_test):
...
@@ -94,6 +94,16 @@ def disabled_tests_onnx_1_8_1(backend_test):
backend_test
.
exclude
(
r
'test_unsqueeze_unsorted_axes_cpu'
)
backend_test
.
exclude
(
r
'test_unsqueeze_unsorted_axes_cpu'
)
def
disabled_tests_onnx_1_10_0
(
backend_test
):
# unsupported shape attributes
backend_test
.
exclude
(
r
'test_shape_end_1_cpu'
)
backend_test
.
exclude
(
r
'test_shape_end_negative_1_cpu'
)
backend_test
.
exclude
(
r
'test_shape_start_1_cpu'
)
backend_test
.
exclude
(
r
'test_shape_start_1_end_2_cpu'
)
backend_test
.
exclude
(
r
'test_shape_start_1_end_negative_1_cpu'
)
backend_test
.
exclude
(
r
'test_shape_start_negative_1_cpu'
)
def
create_backend_test
(
testname
=
None
,
target_device
=
None
):
def
create_backend_test
(
testname
=
None
,
target_device
=
None
):
if
target_device
is
not
None
:
if
target_device
is
not
None
:
c2
.
set_device
(
target_device
)
c2
.
set_device
(
target_device
)
...
@@ -314,6 +324,9 @@ def create_backend_test(testname=None, target_device=None):
...
@@ -314,6 +324,9 @@ def create_backend_test(testname=None, target_device=None):
if
version
.
parse
(
onnx
.
__version__
)
>=
version
.
parse
(
"1.8.0"
):
if
version
.
parse
(
onnx
.
__version__
)
>=
version
.
parse
(
"1.8.0"
):
disabled_tests_onnx_1_8_1
(
backend_test
)
disabled_tests_onnx_1_8_1
(
backend_test
)
if
version
.
parse
(
onnx
.
__version__
)
>=
version
.
parse
(
"1.10.0"
):
disabled_tests_onnx_1_10_0
(
backend_test
)
# import all test cases at global scope to make
# import all test cases at global scope to make
# them visible to python.unittest.
# them visible to python.unittest.
...
...
tools/install_prereqs.sh
View file @
84ce66ec
...
@@ -57,7 +57,7 @@ echo "Dependencies are installed at $PREFIX"
...
@@ -57,7 +57,7 @@ echo "Dependencies are installed at $PREFIX"
rbuild prepare
-d
$PREFIX
-s
develop
rbuild prepare
-d
$PREFIX
-s
develop
# install onnx package for unit tests
# install onnx package for unit tests
pip3
install
onnx
==
1.
8.1
numpy
==
1.21.6
typing
==
3.7.4
pytest
==
6.0.1
packaging
==
16.8
pip3
install
onnx
==
1.
10.0
numpy
==
1.21.6
typing
==
3.7.4
pytest
==
6.0.1
packaging
==
16.8
# pin version of protobuf in Python for onnx runtime unit tests
# pin version of protobuf in Python for onnx runtime unit tests
pip3
install
protobuf
==
3.20.0
pip3
install
protobuf
==
3.20.0
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment