"test/vscode:/vscode.git/clone" did not exist on "ff25a0bbcc8d7b2846a43b53993a1d6023a8a4fb"
Commit 272bce63 authored by Khalique's avatar Khalique
Browse files

added more onnx tests

parent 86663aa5
...@@ -143,29 +143,29 @@ TEST_CASE(constant_test) ...@@ -143,29 +143,29 @@ TEST_CASE(constant_test)
TEST_CASE(constant_fill_test) TEST_CASE(constant_fill_test)
{ {
{
migraphx::program p; migraphx::program p;
auto l0 = p.add_literal(migraphx::literal{{migraphx::shape::int32_type, {2}}, {2, 3}}); migraphx::shape s{migraphx::shape::float_type, {2, 3}};
std::vector<std::size_t> dims(l0->get_shape().elements());
migraphx::literal ls = l0->get_literal();
ls.visit([&](auto s) { dims.assign(s.begin(), s.end()); });
migraphx::shape s{migraphx::shape::float_type, dims};
std::vector<float> value(s.elements(), 1.0); std::vector<float> value(s.elements(), 1.0);
p.add_literal(migraphx::literal{s, value}); p.add_literal(migraphx::literal{s, value});
auto prog = migraphx::parse_onnx("const_fill1_test.onnx"); auto prog = migraphx::parse_onnx("constant_fill_test.onnx");
EXPECT(p == prog); EXPECT(p == prog);
} }
{ TEST_CASE(constant_fill_input_as_shape_test)
{
migraphx::program p; migraphx::program p;
migraphx::shape s{migraphx::shape::float_type, {2, 3}}; auto l0 = p.add_literal(migraphx::literal{{migraphx::shape::int32_type, {2}}, {2, 3}});
std::vector<std::size_t> dims(l0->get_shape().elements());
migraphx::literal ls = l0->get_literal();
ls.visit([&](auto s) { dims.assign(s.begin(), s.end()); });
migraphx::shape s{migraphx::shape::float_type, dims};
std::vector<float> value(s.elements(), 1.0); std::vector<float> value(s.elements(), 1.0);
p.add_literal(migraphx::literal{s, value}); p.add_literal(migraphx::literal{s, value});
auto prog = migraphx::parse_onnx("const_fill2_test.onnx"); auto prog = migraphx::parse_onnx("constant_fill_input_as_shape_test.onnx");
EXPECT(p == prog); EXPECT(p == prog);
}
} }
TEST_CASE(constant_scalar_test) TEST_CASE(constant_scalar_test)
...@@ -358,7 +358,7 @@ TEST_CASE(gemm_ex_test) ...@@ -358,7 +358,7 @@ TEST_CASE(gemm_ex_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(gemm_ex_brcst) TEST_CASE(gemm_ex_brcst_test)
{ {
migraphx::program p; migraphx::program p;
auto l0 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {1, 1, 5, 6}}); auto l0 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {1, 1, 5, 6}});
...@@ -370,7 +370,7 @@ TEST_CASE(gemm_ex_brcst) ...@@ -370,7 +370,7 @@ TEST_CASE(gemm_ex_brcst)
auto alpha = 0.5f; auto alpha = 0.5f;
auto beta = 0.8f; auto beta = 0.8f;
p.add_instruction(migraphx::op::dot{alpha, beta}, t0, l1, t2); p.add_instruction(migraphx::op::dot{alpha, beta}, t0, l1, t2);
auto prog = migraphx::parse_onnx("gemm_test_ex1.onnx"); auto prog = migraphx::parse_onnx("gemm_ex_brcst_test.onnx");
EXPECT(p == prog); EXPECT(p == prog);
} }
...@@ -772,34 +772,34 @@ TEST_CASE(reducemean_keepdims_test) ...@@ -772,34 +772,34 @@ TEST_CASE(reducemean_keepdims_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(reducesum_test1) TEST_CASE(reducesum_test)
{ {
migraphx::program p; migraphx::program p;
auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}}); auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
auto l1 = p.add_instruction(migraphx::op::reduce_sum{{2}}, l0); auto l1 = p.add_instruction(migraphx::op::reduce_sum{{2}}, l0);
p.add_instruction(migraphx::op::squeeze{{2}}, l1); p.add_instruction(migraphx::op::squeeze{{2}}, l1);
auto prog = migraphx::parse_onnx("reducesum_test1.onnx"); auto prog = migraphx::parse_onnx("reducesum_test.onnx");
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(reducesum_test2) TEST_CASE(reducesum_multiaxis_test)
{ {
migraphx::program p; migraphx::program p;
auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}}); auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
auto l1 = p.add_instruction(migraphx::op::reduce_sum{{2, 3}}, l0); auto l1 = p.add_instruction(migraphx::op::reduce_sum{{2, 3}}, l0);
p.add_instruction(migraphx::op::squeeze{{2, 3}}, l1); p.add_instruction(migraphx::op::squeeze{{2, 3}}, l1);
auto prog = migraphx::parse_onnx("reducesum_test2.onnx"); auto prog = migraphx::parse_onnx("reducesum_multiaxis_test.onnx");
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(reducesum_test3) TEST_CASE(reducesum_keepdims_test)
{ {
migraphx::program p; migraphx::program p;
auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}}); auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
p.add_instruction(migraphx::op::reduce_sum{{2, 3}}, l0); p.add_instruction(migraphx::op::reduce_sum{{2, 3}}, l0);
auto prog = migraphx::parse_onnx("reducesum_test3.onnx"); auto prog = migraphx::parse_onnx("reducesum_keepdims_test.onnx");
EXPECT(p == prog); EXPECT(p == prog);
} }
...@@ -820,7 +820,7 @@ TEST_CASE(reshape_test) ...@@ -820,7 +820,7 @@ TEST_CASE(reshape_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(reshape_non_standard) TEST_CASE(reshape_non_standard_test)
{ {
migraphx::program p; migraphx::program p;
migraphx::op::reshape op; migraphx::op::reshape op;
...@@ -830,7 +830,7 @@ TEST_CASE(reshape_non_standard) ...@@ -830,7 +830,7 @@ TEST_CASE(reshape_non_standard)
auto tran_x = p.add_instruction(migraphx::op::transpose{{0, 2, 1}}, x); auto tran_x = p.add_instruction(migraphx::op::transpose{{0, 2, 1}}, x);
auto cont_x = p.add_instruction(migraphx::op::contiguous{}, tran_x); auto cont_x = p.add_instruction(migraphx::op::contiguous{}, tran_x);
p.add_instruction(migraphx::op::reshape{{4, 3, 2}}, cont_x); p.add_instruction(migraphx::op::reshape{{4, 3, 2}}, cont_x);
auto prog = migraphx::parse_onnx("reshape_non_standard.onnx"); auto prog = migraphx::parse_onnx("reshape_non_standard_test.onnx");
EXPECT(p == prog); EXPECT(p == prog);
} }
...@@ -857,7 +857,7 @@ TEST_CASE(shape_gather_test) ...@@ -857,7 +857,7 @@ TEST_CASE(shape_gather_test)
auto l2 = p.add_literal(migraphx::literal{const_shape, {1}}); auto l2 = p.add_literal(migraphx::literal{const_shape, {1}});
int axis = 0; int axis = 0;
p.add_instruction(migraphx::op::gather{axis}, l1, l2); p.add_instruction(migraphx::op::gather{axis}, l1, l2);
auto prog = migraphx::parse_onnx("shape_gather.onnx"); auto prog = migraphx::parse_onnx("shape_gather_test.onnx");
EXPECT(p == prog); EXPECT(p == prog);
} }
......
pow2:u

0
1out"Powpow_testZ
0




Z
1




b
out




B
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