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
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,
std
::
vector
<
shape
::
dynamic_dimension
>
compute_broadcasted_dyn_dims
(
shape
s0
,
shape
s1
)
{
if
(
not
s0
.
dynamic
()
and
not
s1
.
dynamic
())
{
MIGRAPHX_THROW
(
"COMPUTE_BROADCASTED_DYN_DIMS: given two static shapes"
);
}
assert
(
s0
.
dynamic
()
or
s1
.
dynamic
());
// change both shapes to dynamic_dimension representation
if
(
not
s0
.
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():
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
def
binary_dyn_brcst_mul_test
():
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)
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
)
{
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