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
454d9c08
Commit
454d9c08
authored
Oct 19, 2023
by
Umang Yadav
Browse files
Merge branch 'develop' into resnet50_partition
parents
1a92cca5
6e86734d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
2 deletions
+8
-2
tools/accuracy/accuracy_checker.py
tools/accuracy/accuracy_checker.py
+8
-2
No files found.
tools/accuracy/accuracy_checker.py
View file @
454d9c08
...
...
@@ -220,10 +220,16 @@ def main():
else
:
test_input
=
np
.
zeros
(
in_shape
).
astype
(
get_np_datatype
(
in_type
))
test_inputs
[
name
]
=
test_input
params
[
name
]
=
migraphx
.
argument
(
test_input
)
migraphx_arg
=
migraphx
.
argument
(
test_input
)
if
not
args
.
offload_copy
:
migraphx_arg
=
migraphx
.
to_gpu
(
migraphx_arg
)
params
[
name
]
=
migraphx_arg
if
not
args
.
ort_run
:
pred_migx
=
np
.
array
(
model
.
run
(
params
)[
-
1
])
if
not
args
.
offload_copy
:
pred_migx
=
np
.
array
(
migraphx
.
from_gpu
(
model
.
run
(
params
)[
-
1
]))
else
:
pred_migx
=
np
.
array
(
model
.
run
(
params
)[
-
1
])
if
use_onnx
:
sess_op
=
ort
.
SessionOptions
()
...
...
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