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,
...
@@ -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