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
96513f1d
Commit
96513f1d
authored
Aug 29, 2018
by
Paul
Browse files
Formatting
parent
e637b5aa
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
10 deletions
+11
-10
src/onnx/verify_onnx.cpp
src/onnx/verify_onnx.cpp
+11
-10
No files found.
src/onnx/verify_onnx.cpp
View file @
96513f1d
...
...
@@ -14,7 +14,7 @@ auto get_hash(const T& x)
return
std
::
hash
<
T
>
{}(
x
);
}
template
<
class
F
>
template
<
class
F
>
migraph
::
argument
run_cpu
(
F
f
)
{
auto
p
=
f
();
...
...
@@ -29,7 +29,7 @@ migraph::argument run_cpu(F f)
return
out
;
}
template
<
class
F
>
template
<
class
F
>
migraph
::
argument
run_gpu
(
F
f
)
{
auto
p
=
f
();
...
...
@@ -45,8 +45,8 @@ migraph::argument run_gpu(F f)
return
migraph
::
gpu
::
from_gpu
(
out
);
}
template
<
class
F
>
void
verify_program
(
const
std
::
string
&
name
,
F
f
,
double
tolerance
=
100
)
template
<
class
F
>
void
verify_program
(
const
std
::
string
&
name
,
F
f
,
double
tolerance
=
100
)
{
auto
x
=
run_cpu
(
f
);
auto
y
=
run_gpu
(
f
);
...
...
@@ -55,7 +55,7 @@ void verify_program(const std::string & name, F f, double tolerance = 100)
void
verify_instructions
(
const
migraph
::
program
&
prog
,
double
tolerance
=
100
)
{
for
(
auto
&&
ins
:
prog
)
for
(
auto
&&
ins
:
prog
)
{
if
(
ins
.
op
.
name
().
front
()
==
'@'
)
continue
;
...
...
@@ -68,12 +68,13 @@ void verify_instructions(const migraph::program& prog, double tolerance = 100)
auto
create_program
=
[
&
]
{
migraph
::
program
p
;
std
::
vector
<
migraph
::
instruction_ref
>
inputs
;
for
(
auto
&&
arg
:
ins
.
arguments
)
for
(
auto
&&
arg
:
ins
.
arguments
)
{
if
(
arg
->
op
.
name
()
==
"@literal"
)
inputs
.
push_back
(
p
.
add_literal
(
arg
->
lit
));
else
inputs
.
push_back
(
p
.
add_parameter
(
std
::
to_string
(
inputs
.
size
()),
arg
->
get_shape
()));
inputs
.
push_back
(
p
.
add_parameter
(
std
::
to_string
(
inputs
.
size
()),
arg
->
get_shape
()));
}
p
.
add_instruction
(
ins
.
op
,
inputs
);
return
p
;
...
...
@@ -94,7 +95,7 @@ void verify_instructions(const migraph::program& prog, double tolerance = 100)
int
main
(
int
argc
,
char
const
*
argv
[])
{
std
::
vector
<
std
::
string
>
args
(
argv
+
1
,
argv
+
argc
);
std
::
vector
<
std
::
string
>
args
(
argv
+
1
,
argv
+
argc
);
if
(
not
args
.
empty
())
{
std
::
string
file
=
args
.
front
();
...
...
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