add a unit test for better code coverage

test/onnx/onnx_test.cpp

@@ -509,6 +509,22 @@ TEST_CASE(shape_gather_test)
     EXPECT(p == prog);
 }
 
+TEST_CASE(transpose_gather_test)
+{
+    migraphx::program p;
+    auto data  = p.add_parameter("data", migraphx::shape{migraphx::shape::float_type, {3, 5, 4, 6}});
+    auto ind  = p.add_parameter("indices", migraphx::shape{migraphx::shape::int32_type, {2, 4, 3, 5}});
+    auto tr_data = p.add_instruction(migraphx::op::transpose{{0, 2, 1, 3}}, data);
+    auto tr_ind = p.add_instruction(migraphx::op::transpose{{0, 2, 1, 3}}, ind);
+    auto std_data = p.add_instruction(migraphx::op::contiguous{}, tr_data);
+    auto std_ind = p.add_instruction(migraphx::op::contiguous{}, tr_ind);
+    int axis = 1;
+    p.add_instruction(migraphx::op::gather{axis}, std_data, std_ind);
+    auto prog = migraphx::parse_onnx("transpose_gather.onnx");
+
+    EXPECT(p == prog);
+}
+
 TEST_CASE(flatten_test)
 {
     migraphx::program p;