Commit 2338326b authored by Paul's avatar Paul
Browse files

Add explicit call to optimize_onnx

parent acbedb7d
......@@ -12,5 +12,6 @@ endfunction()
file(GLOB ONNX_PARSE_TESTS CONFIGURE_DEPENDS parse/*.cpp)
add_onnx_test(test_onnx_test ${ONNX_PARSE_TESTS})
add_onnx_test(test_onnx_rnn_test onnx_rnn_test.cpp)
add_onnx_test(test_verify_onnx verify_onnx.cpp)
......@@ -4705,7 +4705,7 @@ TEST_CASE(mean_integral_test)
void mvn_n_rank_test(std::vector<int64_t> axes,
std::vector<size_t> input_shape,
const std::string& test_file)
const migraphx::program& prog)
{
using migraphx::make_op;
......@@ -4728,14 +4728,12 @@ void mvn_n_rank_test(std::vector<int64_t> axes,
auto divisor = add_common_op(*mm, make_op("add"), {std, epsilon});
add_common_op(*mm, make_op("div"), {dividend, divisor});
auto prog = optimize_onnx(test_file);
EXPECT(p == prog);
}
TEST_CASE(mvn_default_axes_test)
{
mvn_n_rank_test({0, 2, 3}, {2, 2, 2, 2}, "mvn_default_axes_test.onnx");
mvn_n_rank_test({0, 2, 3}, {2, 2, 2, 2}, optimize_onnx("mvn_default_axes_test.onnx"));
}
TEST_CASE(mvn_default_axes_rank_too_small_test)
......@@ -4749,9 +4747,9 @@ TEST_CASE(mvn_default_axes_rank_too_big_test)
EXPECT(test::throws([&] { migraphx::parse_onnx("mvn_default_axes_rank_too_big_test.onnx"); }));
}
TEST_CASE(mvn_rank_2_test) { mvn_n_rank_test({1}, {2, 2}, "mvn_rank_2_test.onnx"); }
TEST_CASE(mvn_rank_2_test) { mvn_n_rank_test({1}, {2, 2}, optimize_onnx("mvn_rank_2_test.onnx")); }
TEST_CASE(mvn_rank_3_test) { mvn_n_rank_test({0, 1}, {2, 2, 2}, "mvn_rank_3_test.onnx"); }
TEST_CASE(mvn_rank_3_test) { mvn_n_rank_test({0, 1}, {2, 2, 2}, optimize_onnx("mvn_rank_3_test.onnx")); }
TEST_CASE(mvn_axes_rank_too_small_test)
{
......
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