merge with master

src/targets/gpu/target.cpp

@@ -16,7 +16,7 @@
 
 namespace migraph {
 namespace gpu {
-    
+
 std::vector<pass> target::get_passes(migraph::context& gctx) const
 {
     auto& ctx = any_cast<context>(gctx);
@@ -50,9 +50,9 @@ std::string target::name() const { return "miopen"; }
 migraph::context target::get_context(parameter_map params) const
 {
     return context{share(make_obj<miopen_handle>(&miopenCreate)),
-            share(create_rocblas_handle_ptr()), params, {}};
+                   share(create_rocblas_handle_ptr()),
+                   params,
+                   {}};
 }
-
 } // namespace gpu
-
 } // namespace migraph

src/targets/gpu/write_literals.cpp

@@ -26,9 +26,9 @@ struct hip_load_literal
 
 void write_literals::apply(program& p) const
 {
-    if (!enabled(MIGRAPH_DISABLE_MEMORY_COLORING{}))
+    if(!enabled(MIGRAPH_DISABLE_MEMORY_COLORING{}))
         return;
-    
+
     assert(ctx != nullptr);
     for(auto ins : iterator_for(p))
     {
@@ -41,7 +41,5 @@ void write_literals::apply(program& p) const
         }
     }
 }
-
 } // namespace gpu
-
 } // namespace migraph

test/auto_contiguous_test.cpp

@@ -10,7 +10,7 @@ struct contiguous_target
     {
         return {migraph::auto_contiguous{}};
     }
-    migraph::context get_context() const { return {}; }
+    migraph::context get_context(migraph::parameter_map) const { return {}; }
 };
 
 void literal_broadcast()

test/dead_code_elimination_test.cpp

@@ -9,7 +9,7 @@ struct dce_target
     {
         return {migraph::dead_code_elimination{}};
     }
-    migraph::context get_context() const { return {}; }
+    migraph::context get_context(migraph::parameter_map) const { return {}; }
 };
 
 void simple_test()

test/eliminate_allocation_test.cpp

@@ -12,7 +12,7 @@ struct eliminate_allocation_target
     {
         return {migraph::eliminate_allocation{"allocate", align}, migraph::dead_code_elimination{}};
     }
-    migraph::context get_context() const { return {}; }
+    migraph::context get_context(migraph::parameter_map) const { return {}; }
 };
 
 struct allocate

test/memory_coloring_test.cpp

+#include <migraph/memory_coloring.hpp>
+#include <migraph/operators.hpp>
+#include <basic_ops.hpp>
+#include <test.hpp>
+
+struct memory_coloring_target
+{
+    std::string name() const { return "memory_coloring"; }
+    std::vector<migraph::pass> get_passes(migraph::context&) const
+    {
+        return {migraph::memory_coloring{}};
+    }
+    migraph::context get_context(migraph::parameter_map) const { return {}; }
+};
+
+int main()
+{
+    migraph::program p;
+    auto l = p.add_literal(get_2x2());
+    p.add_instruction(migraph::transpose{{1, 0}}, l);
+    p.compile(memory_coloring_target{});
+    EXPECT(p.get_parameter_shape("scratch").bytes() == 16);
+}
+

test/simplify_reshapes_test.cpp

@@ -11,7 +11,7 @@ struct simplify_reshapes_target
     {
         return {migraph::simplify_reshapes{}, migraph::dead_code_elimination{}};
     }
-    migraph::context get_context() const { return {}; }
+    migraph::context get_context(migraph::parameter_map) const { return {}; }
 };
 
 void double_contig()