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
83875c33
Commit
83875c33
authored
Dec 05, 2022
by
Brian Pickrell
Browse files
add Onnx test file
parent
b5a02fea
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
test/onnx/gen_onnx.py
test/onnx/gen_onnx.py
+2
-2
test/onnx/reducel1_dyn_test.onnx
test/onnx/reducel1_dyn_test.onnx
+0
-0
No files found.
test/onnx/gen_onnx.py
View file @
83875c33
...
@@ -23,7 +23,7 @@
...
@@ -23,7 +23,7 @@
#####################################################################################
#####################################################################################
# This script generates onnx files for MIGraphX onnx operator tests.
# This script generates onnx files for MIGraphX onnx operator tests.
# To generate an individual onnx file, you can use the following
# To generate an individual onnx file, you can use the following
# command: python -c "import gen_onnx; gen_onnx.{test_name}_test()"
# command: python
3
-c "import gen_onnx; gen_onnx.{test_name}_test()"
import
numpy
as
np
import
numpy
as
np
import
onnx
import
onnx
from
onnx
import
helper
from
onnx
import
helper
...
@@ -44,7 +44,6 @@ def onnx_test(op_test):
...
@@ -44,7 +44,6 @@ def onnx_test(op_test):
op_info
[
1
],
op_info
[
2
])
op_info
[
1
],
op_info
[
2
])
model_def
=
helper
.
make_model
(
graph_def
,
model_def
=
helper
.
make_model
(
graph_def
,
producer_name
=
op_test
.
__name__
)
producer_name
=
op_test
.
__name__
)
print
(
'hello'
,
op_test
.
__name__
)
onnx
.
save
(
model_def
,
'{}.onnx'
.
format
(
op_test
.
__name__
))
onnx
.
save
(
model_def
,
'{}.onnx'
.
format
(
op_test
.
__name__
))
return
run_test
return
run_test
...
@@ -4452,6 +4451,7 @@ def reducel1_test():
...
@@ -4452,6 +4451,7 @@ def reducel1_test():
return
([
node
],
[
x
],
[
y
])
return
([
node
],
[
x
],
[
y
])
@
onnx_test
def
reducel1_dyn_test
():
def
reducel1_dyn_test
():
x
=
helper
.
make_tensor_value_info
(
'x'
,
TensorProto
.
FLOAT
,
[
None
])
x
=
helper
.
make_tensor_value_info
(
'x'
,
TensorProto
.
FLOAT
,
[
None
])
y
=
helper
.
make_tensor_value_info
(
'y'
,
TensorProto
.
FLOAT
,
[
None
])
y
=
helper
.
make_tensor_value_info
(
'y'
,
TensorProto
.
FLOAT
,
[
None
])
...
...
test/onnx/reducel1_dyn_test.onnx
0 → 100644
View file @
83875c33
File added
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