Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
654fff15
Commit
654fff15
authored
Oct 26, 2022
by
charlie
Browse files
codecov update
parent
58a1da5e
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
37 additions
and
4 deletions
+37
-4
src/common.cpp
src/common.cpp
+1
-4
test/onnx/binary_dyn_brcst_add_test.onnx
test/onnx/binary_dyn_brcst_add_test.onnx
+0
-0
test/onnx/gen_onnx.py
test/onnx/gen_onnx.py
+17
-0
test/onnx/onnx_test.cpp
test/onnx/onnx_test.cpp
+19
-0
No files found.
src/common.cpp
View file @
654fff15
...
@@ -68,10 +68,7 @@ std::vector<std::size_t> compute_broadcasted_lens(std::vector<std::size_t> s0,
...
@@ -68,10 +68,7 @@ std::vector<std::size_t> compute_broadcasted_lens(std::vector<std::size_t> s0,
std
::
vector
<
shape
::
dynamic_dimension
>
compute_broadcasted_dyn_dims
(
shape
s0
,
shape
s1
)
std
::
vector
<
shape
::
dynamic_dimension
>
compute_broadcasted_dyn_dims
(
shape
s0
,
shape
s1
)
{
{
if
(
not
s0
.
dynamic
()
and
not
s1
.
dynamic
())
assert
(
s0
.
dynamic
()
or
s1
.
dynamic
());
{
MIGRAPHX_THROW
(
"COMPUTE_BROADCASTED_DYN_DIMS: given two static shapes"
);
}
// change both shapes to dynamic_dimension representation
// change both shapes to dynamic_dimension representation
if
(
not
s0
.
dynamic
())
if
(
not
s0
.
dynamic
())
s0
=
s0
.
to_dynamic
();
s0
=
s0
.
to_dynamic
();
...
...
test/onnx/binary_dyn_brcst_add_test.onnx
0 → 100644
View file @
654fff15
File added
test/onnx/gen_onnx.py
View file @
654fff15
...
@@ -437,6 +437,23 @@ def binary_dyn_brcst_prelu_test():
...
@@ -437,6 +437,23 @@ def binary_dyn_brcst_prelu_test():
return
([
node
],
[
arg0
,
arg1
],
[
arg_out
])
return
([
node
],
[
arg0
,
arg1
],
[
arg_out
])
@
onnx_test
def
binary_dyn_brcst_add_test
():
arg0
=
helper
.
make_tensor_value_info
(
'0'
,
TensorProto
.
FLOAT16
,
[
4
,
5
])
arg1
=
helper
.
make_tensor_value_info
(
'1'
,
TensorProto
.
FLOAT
,
[
None
,
3
,
4
,
5
])
arg_out
=
helper
.
make_tensor_value_info
(
'out'
,
TensorProto
.
FLOAT
,
[
None
,
3
,
4
,
5
])
node
=
onnx
.
helper
.
make_node
(
'Add'
,
inputs
=
[
'0'
,
'1'
],
outputs
=
[
'out'
],
)
return
([
node
],
[
arg0
,
arg1
],
[
arg_out
])
@
onnx_test
@
onnx_test
def
binary_dyn_brcst_mul_test
():
def
binary_dyn_brcst_mul_test
():
arg0
=
helper
.
make_tensor_value_info
(
'0'
,
TensorProto
.
FLOAT
,
arg0
=
helper
.
make_tensor_value_info
(
'0'
,
TensorProto
.
FLOAT
,
...
...
test/onnx/onnx_test.cpp
View file @
654fff15
...
@@ -541,6 +541,25 @@ TEST_CASE(binary_dyn_brcst_prelu_test)
...
@@ -541,6 +541,25 @@ TEST_CASE(binary_dyn_brcst_prelu_test)
EXPECT
(
p
==
prog
);
EXPECT
(
p
==
prog
);
}
}
TEST_CASE
(
binary_dyn_brcst_add_test
)
{
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
auto
l0
=
mm
->
add_parameter
(
"0"
,
migraphx
::
shape
{
migraphx
::
shape
::
half_type
,
{
4
,
5
}});
auto
l1
=
mm
->
add_parameter
(
"1"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{{
1
,
4
,
0
},
{
3
,
3
,
0
},
{
4
,
4
,
0
},
{
5
,
5
,
0
}}});
auto
ret
=
add_common_op
(
*
mm
,
migraphx
::
make_op
(
"add"
),
{
l0
,
l1
});
mm
->
add_return
({
ret
});
migraphx
::
onnx_options
options
;
options
.
default_dyn_dim_value
=
{
1
,
4
,
0
};
auto
prog
=
migraphx
::
parse_onnx
(
"binary_dyn_brcst_add_test.onnx"
,
options
);
EXPECT
(
p
==
prog
);
}
TEST_CASE
(
binary_dyn_brcst_mul_test
)
TEST_CASE
(
binary_dyn_brcst_mul_test
)
{
{
migraphx
::
program
p
;
migraphx
::
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