add more test for multibroadcast

93d44e6e · Shucai Xiao · 5347c679 · 93d44e6e
Commit 93d44e6e authored Jul 03, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 0 deletions

test/op_shape_test.cpp test/op_shape_test.cpp +10 -0

No files found.
--- a/test/op_shape_test.cpp
+++ b/test/op_shape_test.cpp
@@ -227,6 +227,16 @@ TEST_CASE(multibroadcast)
        migraphx::shape input{migraphx::shape::float_type, {}};
        throws_shape(migraphx::op::multibroadcast{lens}, input);
    }
+    {
+        std::vector<std::size_t> lens{2, 3, 4, 5};
+        migraphx::shape input{migraphx::shape::float_type, {3, 4}};
+        throws_shape(migraphx::op::multibroadcast{lens}, input);
+    }
+    {
+        std::vector<std::size_t> lens{2, 3, 4, 5};
+        migraphx::shape input{migraphx::shape::float_type, {2, 3, 4}};
+        throws_shape(migraphx::op::multibroadcast{lens}, input);
+    }
 }

 TEST_CASE(broadcast)