Merge branch 'multibroadcast_bug' into bert_operators

2445abc3 · Shucai Xiao · 2f6759e7 · 93d44e6e · 2445abc3
Commit 2445abc3 authored Jul 03, 2019 by Shucai Xiao
Show 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)