Update lowering and reshape to work for reshape_copy

Make reshape work like contiguous to perform the copy and then add
proper aliasing in lowering if we're unable to perform a replace instruction

src/include/migraphx/op/reshape.hpp

@@ -46,8 +46,6 @@ struct reshape
         return pack(f(self.dims, "dims"));
     }
 
-    value attributes() const { return {{"require_std_shape", true}}; }
-
     std::string name() const { return "reshape"; }
 
     shape dyn_compute_shape(shape s0) const
@@ -133,17 +131,15 @@ struct reshape
     shape compute_shape(std::vector<shape> inputs) const
     {
         check_shapes{inputs, *this, true}.has(1);
-        auto n_neg_dims = std::count(dims.begin(), dims.end(), -1);
-        if(n_neg_dims > 1)
-            MIGRAPHX_THROW("Reshape: Dimensions for reshape can only have one -1 dim");
-        auto s0 = inputs[0];
+        auto s0 = inputs.front();
         if(s0.dynamic())
         {
-            return dyn_compute_shape(s0);
+            return s0;
         }
         else
         {
-            return static_compute_shape(inputs, n_neg_dims);
+            auto t           = s0.type();
+            return {t, dims};
         } 
     }
 
@@ -152,8 +148,6 @@ struct reshape
         assert(dyn_out.computed_shape.standard());
         argument result{dyn_out.computed_shape};
 
-        //auto resh = args[0](dyn_out.computed_shape);
-
         visit_all(result, args[0])([&](auto output, auto input) {
             shape_for_each(output.get_shape(), [&](const auto& idx) {
                 output(idx.begin(), idx.end()) = input(idx.begin(), idx.end());
@@ -162,7 +156,6 @@ struct reshape
         return result;
     }
 
-    std::ptrdiff_t output_alias(const std::vector<shape>&) const { return 0; }
 };
 
 } // namespace op

src/targets/gpu/lowering.cpp

@@ -90,7 +90,6 @@ struct miopen_apply
         offload_copy   = (mod == mpm->get_root_module()) ? pass->offload_copy : false;
 
         add_generic_op("contiguous");
-        add_generic_op("reshape_lazy");
         add_extend_op("argmax");
         add_extend_op("argmin");
         add_extend_op("logsoftmax");
@@ -116,7 +115,7 @@ struct miopen_apply
         add_neg_op();
         add_nms_op();
         add_select_module_op();
-        //add_reshape_lazy_op();
+        add_reshape_lazy_op();
     }
 
     void copy_params() const
@@ -382,11 +381,11 @@ struct miopen_apply
     /**
      *  Adds reshape lazy to reshape ops that can be aliased instead of copied
      */
-    /*void add_reshape_lazy_op()
+    void add_reshape_lazy_op()
     {
         apply_map.emplace("reshape", [=](instruction_ref ins) {
             /* Attempt lazy reshape to allow for aliasing. Potentially throws in get_shape if unable
-             * to alias 
+             * to alias */
             try
             {
                 auto lazy_ins =  mod->replace_instruction(
@@ -398,10 +397,14 @@ struct miopen_apply
             }
             catch(...)
             {
-                return ins;
+                auto output                       = insert_allocation(ins, ins->get_shape());
+                std::vector<instruction_ref> refs = ins->inputs();
+                refs.push_back(output);
+
+                return mod->replace_instruction(ins, make_op("gpu::contiguous"), refs);
             }
         });
-    }*/
+    }
 };
 
 void lowering::apply(module_pass_manager& mpm) const