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
65f4bf57
Commit
65f4bf57
authored
Nov 24, 2021
by
Shucai Xiao
Browse files
add a unit test
parent
24005e2f
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
25 additions
and
0 deletions
+25
-0
test/verify/test_mul_lrn.cpp
test/verify/test_mul_lrn.cpp
+25
-0
No files found.
test/verify/test_mul_lrn.cpp
0 → 100644
View file @
65f4bf57
#include "verify_program.hpp"
#include <migraphx/program.hpp>
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
struct
test_mul_lrn
:
verify_program
<
test_mul_lrn
>
{
migraphx
::
program
create_program
()
const
{
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
migraphx
::
shape
s
{
migraphx
::
shape
::
half_type
,
{
1
,
5
,
2
,
2
}};
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
));
auto
xl
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"mul"
),
x
,
l
);
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
}}),
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