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
bad02557
Commit
bad02557
authored
Nov 24, 2021
by
Shucai Xiao
Browse files
refine the unit test
parent
7ac9c87b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
test/verify/test_mul_lrn.cpp
test/verify/test_mul_lrn.cpp
+2
-2
No files found.
test/verify/test_mul_lrn.cpp
View file @
bad02557
...
...
@@ -10,7 +10,7 @@ struct test_mul_lrn : verify_program<test_mul_lrn>
{
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
migraphx
::
shape
s
{
migraphx
::
shape
::
half_type
,
{
1
,
5
,
2
,
2
}};
migraphx
::
shape
s
{
migraphx
::
shape
::
half_type
,
{
1
,
3
,
2
24
,
2
24
}};
auto
x
=
mm
->
add_parameter
(
"x"
,
s
);
std
::
vector
<
float
>
vec
(
s
.
elements
(),
500.0
f
);
auto
l
=
mm
->
add_literal
(
migraphx
::
literal
(
s
,
vec
));
...
...
@@ -18,7 +18,7 @@ struct test_mul_lrn : verify_program<test_mul_lrn>
auto
y
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"relu"
),
xl
);
mm
->
add_instruction
(
migraphx
::
make_op
(
"lrn"
,
{{
"alpha"
,
0.0001
},
{
"beta"
,
0.75
},
{
"bias"
,
1.0
},
{
"size"
,
5
}}),
{{
"alpha"
,
0.0001
},
{
"beta"
,
0.75
},
{
"bias"
,
1.0
},
{
"size"
,
5
0
}}),
y
);
return
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