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
a79ab4d7
Commit
a79ab4d7
authored
Nov 12, 2018
by
Khalique
Browse files
formatting
parent
5168b178
Changes
5
Show whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
24 additions
and
29 deletions
+24
-29
src/targets/cpu/lowering.cpp
src/targets/cpu/lowering.cpp
+7
-7
src/targets/gpu/LRN.cpp
src/targets/gpu/LRN.cpp
+11
-11
src/targets/gpu/include/migraph/gpu/miopen.hpp
src/targets/gpu/include/migraph/gpu/miopen.hpp
+1
-6
src/targets/gpu/lowering.cpp
src/targets/gpu/lowering.cpp
+4
-3
test/cpu_ops_test.cpp
test/cpu_ops_test.cpp
+1
-2
No files found.
src/targets/cpu/lowering.cpp
View file @
a79ab4d7
src/targets/gpu/LRN.cpp
View file @
a79ab4d7
src/targets/gpu/include/migraph/gpu/miopen.hpp
View file @
a79ab4d7
...
...
@@ -88,12 +88,7 @@ inline pooling_descriptor make_pooling(const migraph::op::pooling& op)
inline
LRN_descriptor
make_LRN
(
const
migraph
::
op
::
LRN
&
op
)
{
auto
ldesc
=
make_obj
<
LRN_descriptor
>
(
&
miopenCreateLRNDescriptor
);
miopenSetLRNDescriptor
(
ldesc
.
get
(),
miopenLRNCrossChannel
,
op
.
size
,
op
.
alpha
,
op
.
beta
,
op
.
bias
);
miopenSetLRNDescriptor
(
ldesc
.
get
(),
miopenLRNCrossChannel
,
op
.
size
,
op
.
alpha
,
op
.
beta
,
op
.
bias
);
return
ldesc
;
}
...
...
src/targets/gpu/lowering.cpp
View file @
a79ab4d7
...
...
@@ -142,7 +142,8 @@ struct miopen_apply
auto
&&
op
=
any_cast
<
op
::
LRN
>
(
ins
->
get_operator
());
auto
ldesc
=
make_LRN
(
op
);
auto
output
=
insert_allocation
(
ins
,
ins
->
get_shape
());
return
prog
->
replace_instruction
(
ins
,
miopen_LRN
{
std
::
move
(
ldesc
)},
ins
->
inputs
().
at
(
0
),
output
);
return
prog
->
replace_instruction
(
ins
,
miopen_LRN
{
std
::
move
(
ldesc
)},
ins
->
inputs
().
at
(
0
),
output
);
}
instruction_ref
apply_relu
(
instruction_ref
ins
)
...
...
test/cpu_ops_test.cpp
View file @
a79ab4d7
...
...
@@ -589,11 +589,10 @@ TEST_CASE(LRN_test)
auto
result
=
p
.
eval
({});
std
::
vector
<
float
>
results_vector
(
5
);
result
.
visit
([
&
](
auto
output
)
{
results_vector
.
assign
(
output
.
begin
(),
output
.
end
());
});
std
::
vector
<
float
>
gold
=
{
-
2
/
1.000075
,
1
/
1.00009
,
0
/
1.000145
,
1
/
1.00009
,
2
/
1.000075
};
std
::
vector
<
float
>
gold
=
{
-
2
/
1.000075
,
1
/
1.00009
,
0
/
1.000145
,
1
/
1.00009
,
2
/
1.000075
};
EXPECT
(
migraph
::
verify_range
(
results_vector
,
gold
));
}
TEST_CASE
(
imagescaler_test
)
{
migraph
::
program
p
;
...
...
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