Skip to content

GitLab

Commit f1bce841 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format.

parent 09336f53
Show whitespace changes
Inline Side-by-side
Showing with 105 additions and 77 deletions
test/gpu/miopen.cpp
View file @ f1bce841
...@@ -1073,7 +1073,8 @@ struct test_rnn_forward ...@@ -1073,7 +1073,8 @@ struct test_rnn_forward
auto r = p.add_parameter("r", r_shape); auto r = p.add_parameter("r", r_shape);
auto bias = p.add_parameter("bias", b_shape); auto bias = p.add_parameter("bias", b_shape);
auto output = p.add_instruction(migraphx::op::rnn{hidden_size, auto output =
p.add_instruction(migraphx::op::rnn{hidden_size,
{migraphx::op::tanh{}, migraphx::op::tanh{}}, {migraphx::op::tanh{}, migraphx::op::tanh{}},
migraphx::op::rnn::forward, migraphx::op::rnn::forward,
clip}, clip},
...@@ -1113,7 +1114,8 @@ struct test_rnn_reverse ...@@ -1113,7 +1114,8 @@ struct test_rnn_reverse
auto r = p.add_parameter("r", r_shape); auto r = p.add_parameter("r", r_shape);
auto bias = p.add_parameter("bias", b_shape); auto bias = p.add_parameter("bias", b_shape);
auto output = p.add_instruction(migraphx::op::rnn{hidden_size, auto output =
p.add_instruction(migraphx::op::rnn{hidden_size,
{migraphx::op::tanh{}, migraphx::op::tanh{}}, {migraphx::op::tanh{}, migraphx::op::tanh{}},
migraphx::op::rnn::reverse, migraphx::op::rnn::reverse,
clip}, clip},
...@@ -1153,7 +1155,8 @@ struct test_rnn_bidirectional ...@@ -1153,7 +1155,8 @@ struct test_rnn_bidirectional
auto r = p.add_parameter("r", r_shape); auto r = p.add_parameter("r", r_shape);
auto bias = p.add_parameter("bias", b_shape); auto bias = p.add_parameter("bias", b_shape);
auto output = p.add_instruction(migraphx::op::rnn{hidden_size, auto output =
p.add_instruction(migraphx::op::rnn{hidden_size,
{migraphx::op::tanh{}, migraphx::op::tanh{}}, {migraphx::op::tanh{}, migraphx::op::tanh{}},
migraphx::op::rnn::bidirectional, migraphx::op::rnn::bidirectional,
clip}, clip},
......
test/op_shape_test.cpp
View file @ f1bce841
...@@ -266,11 +266,15 @@ TEST_CASE(rnn) ...@@ -266,11 +266,15 @@ TEST_CASE(rnn)
migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}}; migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}};
migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}}; migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}};
expect_shape(migraphx::shape{migraphx::shape::float_type, {seq_len, num_dirct, batch_size, hidden_size}}, expect_shape(migraphx::shape{migraphx::shape::float_type,
migraphx::op::rnn{hidden_size, {seq_len, num_dirct, batch_size, hidden_size}},
{migraphx::op::tanh{}}, migraphx::op::rnn{
migraphx::op::rnn::forward, clip}, hidden_size, {migraphx::op::tanh{}}, migraphx::op::rnn::forward, clip},
in_shape, w_shape, r_shape, b_shape, ih_shape); in_shape,
w_shape,
r_shape,
b_shape,
ih_shape);
} }
{ {
...@@ -288,11 +292,15 @@ TEST_CASE(rnn) ...@@ -288,11 +292,15 @@ TEST_CASE(rnn)
migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}}; migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}};
migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}}; migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}};
expect_shape(migraphx::shape{migraphx::shape::float_type, {seq_len, num_dirct, batch_size, hidden_size}}, expect_shape(migraphx::shape{migraphx::shape::float_type,
migraphx::op::rnn{hidden_size, {seq_len, num_dirct, batch_size, hidden_size}},
{migraphx::op::tanh{}}, migraphx::op::rnn{
migraphx::op::rnn::reverse, clip}, hidden_size, {migraphx::op::tanh{}}, migraphx::op::rnn::reverse, clip},
in_shape, w_shape, r_shape, b_shape, ih_shape); in_shape,
w_shape,
r_shape,
b_shape,
ih_shape);
} }
{ {
...@@ -310,11 +318,16 @@ TEST_CASE(rnn) ...@@ -310,11 +318,16 @@ TEST_CASE(rnn)
migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}}; migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}};
migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}}; migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}};
expect_shape(migraphx::shape{migraphx::shape::float_type, {seq_len, num_dirct, batch_size, hidden_size}}, expect_shape(
migraphx::op::rnn{hidden_size, migraphx::shape{migraphx::shape::float_type,
{migraphx::op::tanh{}}, {seq_len, num_dirct, batch_size, hidden_size}},
migraphx::op::rnn::bidirectional, clip}, migraphx::op::rnn{
in_shape, w_shape, r_shape, b_shape, ih_shape); hidden_size, {migraphx::op::tanh{}}, migraphx::op::rnn::bidirectional, clip},
in_shape,
w_shape,
r_shape,
b_shape,
ih_shape);
} }
{ {
...@@ -332,10 +345,14 @@ TEST_CASE(rnn) ...@@ -332,10 +345,14 @@ TEST_CASE(rnn)
migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}}; migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}};
migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}}; migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}};
throws_shape(migraphx::op::rnn{hidden_size + 1, throws_shape(
{migraphx::op::tanh{}}, migraphx::op::rnn{
migraphx::op::rnn::forward, clip}, hidden_size + 1, {migraphx::op::tanh{}}, migraphx::op::rnn::forward, clip},
in_shape, w_shape, r_shape, b_shape, ih_shape); in_shape,
w_shape,
r_shape,
b_shape,
ih_shape);
} }
{ {
...@@ -353,10 +370,14 @@ TEST_CASE(rnn) ...@@ -353,10 +370,14 @@ TEST_CASE(rnn)
migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}}; migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}};
migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}}; migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}};
throws_shape(migraphx::op::rnn{hidden_size, throws_shape(
{migraphx::op::tanh{}}, migraphx::op::rnn{
migraphx::op::rnn::bidirectional, clip}, hidden_size, {migraphx::op::tanh{}}, migraphx::op::rnn::bidirectional, clip},
in_shape, w_shape, r_shape, b_shape, ih_shape); in_shape,
w_shape,
r_shape,
b_shape,
ih_shape);
} }
{ {
...@@ -374,10 +395,14 @@ TEST_CASE(rnn) ...@@ -374,10 +395,14 @@ TEST_CASE(rnn)
migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}}; migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}};
migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}}; migraphx::shape b_shape{migraphx::shape::float_type, {num_dirct, 2 * hidden_size}};
throws_shape(migraphx::op::rnn{hidden_size, throws_shape(
{migraphx::op::tanh{}}, migraphx::op::rnn{
migraphx::op::rnn::forward, clip}, hidden_size, {migraphx::op::tanh{}}, migraphx::op::rnn::forward, clip},
in_shape, w_shape, r_shape, b_shape, ih_shape); in_shape,
w_shape,
r_shape,
b_shape,
ih_shape);
} }
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment