Commit e83dc134 authored by Paul's avatar Paul
Browse files

Dont hinder eliminate_contiguous

parent 8e49a9f2
...@@ -25,6 +25,7 @@ ...@@ -25,6 +25,7 @@
#include <migraphx/instruction.hpp> #include <migraphx/instruction.hpp>
#include <migraphx/register_op.hpp> #include <migraphx/register_op.hpp>
#include <migraphx/array.hpp> #include <migraphx/array.hpp>
#include <migraphx/make_op.hpp>
#include <migraphx/op/clip.hpp> #include <migraphx/op/clip.hpp>
#include <cmath> #include <cmath>
#include <set> #include <set>
...@@ -948,6 +949,22 @@ struct find_commutative_broadcast ...@@ -948,6 +949,22 @@ struct find_commutative_broadcast
} }
}; };
struct find_contiguous
{
auto matcher() const
{
return match::name("gpu::contiguous");
}
void apply(module& m, const match::matcher_result& r) const
{
auto ins = r.result;
auto args = ins->inputs();
m.replace_instruction(ins, make_op("gpu::precompile_op", {{"op", to_value(make_op("contiguous"))}}), ins->inputs());
}
};
void fuse_ops::apply(module& m) const void fuse_ops::apply(module& m) const
{ {
match::find_matches(m, find_gelu{}, find_gelu_new{fast_math}); match::find_matches(m, find_gelu{}, find_gelu_new{fast_math});
...@@ -968,6 +985,7 @@ void fuse_ops::apply(module& m) const ...@@ -968,6 +985,7 @@ void fuse_ops::apply(module& m) const
find_add_clip{}); find_add_clip{});
run_passes(m, {dead_code_elimination{}}); run_passes(m, {dead_code_elimination{}});
match::find_matches(m, find_triadd_layernorm{}, find_gemm_add{}, find_commutative_broadcast{}); match::find_matches(m, find_triadd_layernorm{}, find_gemm_add{}, find_commutative_broadcast{});
match::find_matches(m, find_contiguous{});
} }
} // namespace gpu } // namespace gpu
......
...@@ -130,7 +130,7 @@ struct miopen_apply ...@@ -130,7 +130,7 @@ struct miopen_apply
add_generic_op("atan"); add_generic_op("atan");
add_generic_op("atanh"); add_generic_op("atanh");
add_generic_op("ceil"); add_generic_op("ceil");
// add_generic_op("contiguous"); add_generic_op("contiguous");
add_generic_op("cos"); add_generic_op("cos");
add_generic_op("cosh"); add_generic_op("cosh");
add_generic_op("div"); add_generic_op("div");
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment