Update tests, possible throw that won't occur

src/include/migraphx/op/broadcast.hpp

@@ -86,6 +86,7 @@ struct broadcast
             shape output{t, broadcast_lens, std::move(bcast_strides)};
             if(output.elements() < s0.elements())
             {
+                // don't think this can occur?
                 MIGRAPHX_THROW("BROADCAST: output size must be greater than or equal to s0 size");
             }
             return output;
@@ -125,9 +126,6 @@ struct broadcast
             std::vector<size_t> bcast_strides(s1.ndim(), 0);
             std::copy(s0.strides().begin(), s0.strides().end(), bcast_strides.begin() + axis);
             shape output{t, s1.lens(), std::move(bcast_strides)};
-            if(output.elements() < s0.elements())
-                MIGRAPHX_THROW(
-                    "BROADCAST_2in: output size must be greater than or equal to s0 size");
             return output;
         }
     }

test/op_shape_test.cpp

@@ -118,9 +118,15 @@ TEST_CASE(broadcast)
     }
 }
 
-TEST_CASE(broadcast_2in)
+TEST_CASE(broadcast_axis_out_of_range_error)
+{
+    std::vector<std::size_t> lens{1, 1};
+    migraphx::shape input{migraphx::shape::float_type, {1}, {0}};
+    throws_shape(migraphx::make_op("broadcast", {{"axis", 4}, {"out_lens", lens}}), input);
+}
+
+TEST_CASE(broadcast_2in_static_static)
 {
-    {
     migraphx::shape a_input{migraphx::shape::float_type, {4}, {1}};
     migraphx::shape b_input{migraphx::shape::float_type, {4, 4}, {4, 1}};
     expect_shape(migraphx::shape{migraphx::shape::float_type, {4, 4}, {1, 0}},
@@ -132,34 +138,47 @@ TEST_CASE(broadcast_2in)
                  a_input,
                  b_input);
     throws_shape(migraphx::make_op("broadcast", {{"axis", 2}}), a_input, b_input);
-    }
-    {
+}
+
+TEST_CASE(broadcast_2in_not_matching_error)
+{
     migraphx::shape a_input{migraphx::shape::float_type, {4}, {1}};
     migraphx::shape b_input{migraphx::shape::float_type, {2, 2}, {2, 1}};
     throws_shape(migraphx::make_op("broadcast", {{"axis", 1}}), a_input, b_input);
-    }
-    {
+}
+
+TEST_CASE(broadcast_2in_dynamic_s0_error1)
+{
     migraphx::shape a_input{migraphx::shape::float_type, {4, 2}, {2, 1}};
     migraphx::shape b_input{migraphx::shape::float_type, {{1, 4, 0}, {4, 4, 0}, {2, 2, 0}}};
     throws_shape(migraphx::make_op("broadcast", {{"axis", 0}}), b_input, a_input);
-    }
-    {
+}
+
+TEST_CASE(broadcast_2in_dynamic_s0_error2)
+{
     std::vector<migraphx::shape::dynamic_dimension> dd{{4, 4, 0}};
     migraphx::shape a_input{migraphx::shape::float_type, dd};
     migraphx::shape b_input{migraphx::shape::float_type, {4, 4}, {4, 1}};
     throws_shape(migraphx::make_op("broadcast", {{"axis", 0}}), a_input, b_input);
-    }
-    {
+}
+
+TEST_CASE(broadcast_2in_static_dyn)
+{
     migraphx::shape a_input{migraphx::shape::float_type, {4}, {1}};
     migraphx::shape b_input{migraphx::shape::float_type, {{1, 4, 0}, {4, 4, 0}, {2, 2, 0}}};
     throws_shape(migraphx::make_op("broadcast", {{"axis", 0}}), a_input, b_input);
-        expect_shape(
-            migraphx::shape{migraphx::shape::float_type, {{1, 4, 0}, {4, 4, 0}, {2, 2, 0}}},
+    expect_shape(migraphx::shape{migraphx::shape::float_type, {{1, 4, 0}, {4, 4, 0}, {2, 2, 0}}},
                  migraphx::make_op("broadcast", {{"axis", 1}}),
                  a_input,
                  b_input);
     throws_shape(migraphx::make_op("broadcast", {{"axis", 2}}), a_input, b_input);
-    }
+}
+
+TEST_CASE(broadcast_2in_dyn_s0_ndim_greater_than_1_error)
+{
+    migraphx::shape a_input{migraphx::shape::float_type, {4, 2}};
+    migraphx::shape b_input{migraphx::shape::float_type, {{1, 4, 0}, {4, 4, 0}, {2, 2, 0}}};
+    throws_shape(migraphx::make_op("broadcast", {{"axis", 0}}), a_input, b_input);
 }
 
 TEST_CASE(convolution_shape)