Code cov update

src/onnx/parse_gemm.cpp

@@ -44,7 +44,7 @@ struct parse_gemm : op_parser<parse_gemm>
         if(a_arg->get_shape().ndim() != 2 or b_arg->get_shape().ndim() != 2)
         {
             MIGRAPHX_THROW("PARSE_GEMM: A and B should be rank 2, A is rank " +
-                           std::to_string(a_arg->get_shape().ndim()) + "B is rank " +
+                           std::to_string(a_arg->get_shape().ndim()) + ", B is rank " +
                            std::to_string(b_arg->get_shape().ndim()));
         }
 

test/onnx/gen_onnx.py

@@ -2103,6 +2103,23 @@ def gemm_dyn_C_error():
     return ([node], [A, B, C], [Y])
 
 
+@onnx_test
+def gemm_rank_error():
+    A = helper.make_tensor_value_info('A', TensorProto.FLOAT, [4, 1, 8, 6])
+    B = helper.make_tensor_value_info('B', TensorProto.FLOAT, [4, 1, 8, 7])
+    C = helper.make_tensor_value_info('C', TensorProto.FLOAT, [6, 7])
+    Y = helper.make_tensor_value_info('Y', TensorProto.FLOAT, [4, 1, 6, 7])
+
+    node = onnx.helper.make_node('Gemm',
+                                 inputs=['A', 'B', 'C'],
+                                 outputs=['Y'],
+                                 alpha=0.5,
+                                 beta=0.8,
+                                 transA=1)
+
+    return ([node], [A, B, C], [Y])
+
+
 @onnx_test
 def globalavgpool_test():
     x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 16, 16])

test/onnx/onnx_test.cpp

@@ -2176,6 +2176,11 @@ TEST_CASE(gemm_dyn_C_error)
     EXPECT(test::throws([&] { migraphx::parse_onnx("gemm_dyn_C_error.onnx", options); }));
 }
 
+TEST_CASE(gemm_rank_error)
+{
+    EXPECT(test::throws([&] { migraphx::parse_onnx("gemm_rank_error.onnx"); }));
+}
+
 TEST_CASE(globalavgpool_test)
 {
     migraphx::program p;