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
d2fa123c
Commit
d2fa123c
authored
Oct 24, 2022
by
charlie
Browse files
Add ref ops test
parent
5120f911
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
32 additions
and
3 deletions
+32
-3
test/ref_ops_test.cpp
test/ref_ops_test.cpp
+32
-3
No files found.
test/ref_ops_test.cpp
View file @
d2fa123c
...
...
@@ -1948,6 +1948,35 @@ TEST_CASE(equal_test)
EXPECT
(
results_vector
==
gold
);
}
TEST_CASE
(
equal_dynamic_test
)
{
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
std
::
vector
<
migraphx
::
shape
::
dynamic_dimension
>
dd
{{
6
,
12
,
9
}};
migraphx
::
shape
s
{
migraphx
::
shape
::
float_type
,
dd
};
auto
p0
=
mm
->
add_parameter
(
"l"
,
s
);
auto
p1
=
mm
->
add_parameter
(
"r"
,
s
);
auto
eq
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"equal"
),
p0
,
p1
);
auto
r
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"convert"
,
{{
"target_type"
,
migraphx
::
to_value
(
migraphx
::
shape
::
bool_type
)}}),
eq
);
mm
->
add_return
({
r
});
p
.
compile
(
migraphx
::
ref
::
target
{});
std
::
vector
<
float
>
l_data
{
1.1
,
1.5
,
0.1
,
-
1.1
,
-
1.5
,
-
0.6
,
0.0
,
2.0
,
-
2.0
};
std
::
vector
<
float
>
r_data
{
1.1
,
1.6
,
-
0.1
,
-
1.2
,
-
1.5
,
-
0.7
,
0.0
,
2.3
,
-
2.1
};
migraphx
::
parameter_map
params0
;
migraphx
::
shape
input_fixed_shape0
{
migraphx
::
shape
::
float_type
,
{
9
}};
params0
[
"x"
]
=
migraphx
::
argument
(
input_fixed_shape0
,
l_data
.
data
());
params0
[
"slope"
]
=
migraphx
::
argument
(
input_fixed_shape0
,
r_data
.
data
());
auto
result
=
p
.
eval
(
params0
).
back
();
std
::
vector
<
bool
>
results_vector
;
result
.
visit
([
&
](
auto
output
)
{
results_vector
.
assign
(
output
.
begin
(),
output
.
end
());
});
std
::
vector
<
bool
>
gold
=
{
true
,
false
,
false
,
false
,
true
,
false
,
true
,
false
,
false
};
EXPECT
(
results_vector
==
gold
);
}
TEST_CASE
(
erf_test
)
{
migraphx
::
program
p
;
...
...
@@ -4768,15 +4797,15 @@ TEST_CASE(prelu_dynamic_test)
{
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
std
::
vector
<
migraphx
::
shape
::
dynamic_dimension
>
dd
{{
3
,
3
,
0
}};
std
::
vector
<
migraphx
::
shape
::
dynamic_dimension
>
dd
{{
2
,
6
,
0
}};
migraphx
::
shape
s
{
migraphx
::
shape
::
float_type
,
dd
};
std
::
vector
<
float
>
x_data
{
-
1
,
0
,
2
};
std
::
vector
<
float
>
slope_data
{
2
,
1
,
2
};
auto
x
=
mm
->
add_parameter
(
"x"
,
s
);
auto
slope
=
mm
->
add_parameter
(
"slope"
,
s
);
mm
->
add_instruction
(
migraphx
::
make_op
(
"prelu"
),
x
,
slope
);
p
.
compile
(
migraphx
::
ref
::
target
{});
std
::
vector
<
float
>
x_data
{
-
1
,
0
,
2
};
std
::
vector
<
float
>
slope_data
{
2
,
1
,
2
};
migraphx
::
parameter_map
params0
;
migraphx
::
shape
input_fixed_shape0
{
migraphx
::
shape
::
float_type
,
{
3
}};
params0
[
"x"
]
=
migraphx
::
argument
(
input_fixed_shape0
,
x_data
.
data
());
...
...
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