Fix test

cdf1d239 · charlie · 4d59204f · cdf1d239
Commit cdf1d239 authored Nov 17, 2022 by charlie
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 3 deletions

test/onnx/onnx_test.cpp test/onnx/onnx_test.cpp +5 -3

No files found.
--- a/test/onnx/onnx_test.cpp
+++ b/test/onnx/onnx_test.cpp
@@ -5699,9 +5699,11 @@ TEST_CASE(squeeze_unsqueeze_dyn_test)
    auto l0 = mm->add_parameter(
        "0",
        migraphx::shape{migraphx::shape::float_type,
-                        {{1, 1, 0}, {1, 4, 0}, {1, 1, 0}, {1, 1, 1}, {1, 4, 0}, {1, 1, 0}}});
+                        {{1, 1, 0}, {1, 4, 0}, {1, 1, 0}, {1, 1, 0}, {1, 4, 0}, {1, 1, 0}}});
-    auto l1  = mm->add_instruction(migraphx::make_op("squeeze", {{"axes", squeeze_axes}}), l0);
+    auto c0  = mm->add_instruction(migraphx::make_op("contiguous"), l0);
-    auto ret = mm->add_instruction(migraphx::make_op("unsqueeze", {{"axes", unsqueeze_axes}}), l1);
+    auto l1  = mm->add_instruction(migraphx::make_op("squeeze", {{"axes", squeeze_axes}}), c0);
+    auto c1  = mm->add_instruction(migraphx::make_op("contiguous"), l1);
+    auto ret = mm->add_instruction(migraphx::make_op("unsqueeze", {{"axes", unsqueeze_axes}}), c1);
    mm->add_return({ret});
    migraphx::onnx_options options;