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
2338326b
Commit
2338326b
authored
Dec 13, 2023
by
Paul
Browse files
Add explicit call to optimize_onnx
parent
acbedb7d
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
6 deletions
+5
-6
test/onnx/CMakeLists.txt
test/onnx/CMakeLists.txt
+1
-0
test/onnx/onnx_test.cpp
test/onnx/onnx_test.cpp
+4
-6
No files found.
test/onnx/CMakeLists.txt
View file @
2338326b
...
...
@@ -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
)
test/onnx/onnx_test.cpp
View file @
2338326b
...
...
@@ -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)
{
...
...
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