Merge branch 'develop' into finalize

test/op_shape_test.cpp

@@ -212,4 +212,24 @@ TEST_CASE(multibroadcast)
     }
 }
 
+TEST_CASE(gather)
+{
+    {
+        migraphx::shape input{migraphx::shape::float_type, {2, 3, 4, 5}};
+        migraphx::shape indices{migraphx::shape::int32_type, {2, 3}};
+        std::size_t axis = 1;
+        expect_shape(migraphx::shape{migraphx::shape::float_type, {2, 6, 4, 5}},
+                     migraphx::op::gather{axis},
+                     input,
+                     indices);
+    }
+
+    {
+        migraphx::shape input{migraphx::shape::float_type, {2, 3, 4, 5}};
+        migraphx::shape indices{migraphx::shape::int32_type, {2, 3}};
+        std::size_t axis = 4;
+        throws_shape(migraphx::op::gather{axis}, input, indices);
+    }
+}
+
 int main(int argc, const char* argv[]) { test::run(argc, argv); }