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
83b91229
Commit
83b91229
authored
Jun 29, 2018
by
Paul
Browse files
Formatting
parent
8e0fff81
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
15 additions
and
15 deletions
+15
-15
src/include/rtg/operators.hpp
src/include/rtg/operators.hpp
+14
-13
src/onnx/verify_onnx.cpp
src/onnx/verify_onnx.cpp
+1
-2
No files found.
src/include/rtg/operators.hpp
View file @
83b91229
...
@@ -16,21 +16,25 @@ struct check_shapes
...
@@ -16,21 +16,25 @@ struct check_shapes
check_shapes
(
const
std
::
vector
<
shape
>&
s
)
:
shapes
(
&
s
)
{}
check_shapes
(
const
std
::
vector
<
shape
>&
s
)
:
shapes
(
&
s
)
{}
template
<
class
Op
>
template
<
class
Op
>
check_shapes
(
const
std
::
vector
<
shape
>&
s
,
const
Op
&
op
)
:
shapes
(
&
s
),
name
(
op
.
name
())
{}
check_shapes
(
const
std
::
vector
<
shape
>&
s
,
const
Op
&
op
)
:
shapes
(
&
s
),
name
(
op
.
name
())
{
}
std
::
string
prefix
()
const
std
::
string
prefix
()
const
{
{
if
(
name
.
empty
())
return
""
;
if
(
name
.
empty
())
else
return
name
+
": "
;
return
""
;
else
return
name
+
": "
;
}
}
const
check_shapes
&
has
(
std
::
size_t
n
)
const
const
check_shapes
&
has
(
std
::
size_t
n
)
const
{
{
assert
(
shapes
!=
nullptr
);
assert
(
shapes
!=
nullptr
);
if
(
shapes
->
size
()
!=
n
)
if
(
shapes
->
size
()
!=
n
)
RTG_THROW
(
prefix
()
+
"Wrong number of arguments: expected "
+
std
::
to_string
(
n
)
+
" but given "
+
RTG_THROW
(
prefix
()
+
"Wrong number of arguments: expected "
+
std
::
to_string
(
n
)
+
std
::
to_string
(
shapes
->
size
()));
" but given "
+
std
::
to_string
(
shapes
->
size
()));
return
*
this
;
return
*
this
;
}
}
...
@@ -508,14 +512,11 @@ struct outline
...
@@ -508,14 +512,11 @@ struct outline
argument
compute
(
context
&
,
shape
,
std
::
vector
<
argument
>
)
const
{
return
{
s
,
nullptr
};
}
argument
compute
(
context
&
,
shape
,
std
::
vector
<
argument
>
)
const
{
return
{
s
,
nullptr
};
}
};
};
template
<
class
T
>
template
<
class
T
>
struct
check_context
struct
check_context
{
{
std
::
string
name
()
const
{
return
"check_context"
;
}
std
::
string
name
()
const
{
return
"check_context"
;
}
shape
compute_shape
(
std
::
vector
<
shape
>
)
const
shape
compute_shape
(
std
::
vector
<
shape
>
)
const
{
return
{};
}
{
return
{};
}
argument
compute
(
context
&
ctx
,
shape
,
std
::
vector
<
argument
>
)
const
argument
compute
(
context
&
ctx
,
shape
,
std
::
vector
<
argument
>
)
const
{
{
T
*
x
=
any_cast
<
T
>
(
&
ctx
);
T
*
x
=
any_cast
<
T
>
(
&
ctx
);
...
...
src/onnx/verify_onnx.cpp
View file @
83b91229
...
@@ -29,8 +29,7 @@ rtg::argument run_gpu(std::string file)
...
@@ -29,8 +29,7 @@ rtg::argument run_gpu(std::string file)
auto
output
=
rtg
::
miopen
::
to_gpu
(
rtg
::
generate_argument
(
p
.
get_parameter_shape
(
"output"
)));
auto
output
=
rtg
::
miopen
::
to_gpu
(
rtg
::
generate_argument
(
p
.
get_parameter_shape
(
"output"
)));
auto
handle
=
rtg
::
miopen
::
make_obj
<
rtg
::
miopen
::
miopen_handle
>
(
&
miopenCreate
);
auto
handle
=
rtg
::
miopen
::
make_obj
<
rtg
::
miopen
::
miopen_handle
>
(
&
miopenCreate
);
auto
out
=
p
.
eval
(
auto
out
=
p
.
eval
({{
"Input3"
,
input3
},
{
"output"
,
output
}});
{{
"Input3"
,
input3
},
{
"output"
,
output
}});
std
::
cout
<<
p
<<
std
::
endl
;
std
::
cout
<<
p
<<
std
::
endl
;
return
rtg
::
miopen
::
from_gpu
(
out
);
return
rtg
::
miopen
::
from_gpu
(
out
);
}
}
...
...
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