"...composable_kernel_onnxruntime.git" did not exist on "3cb2a7d09ff4f594c2ca5532d790b845ecb681c9"
Commit 8b28251b authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

update to replace module inputs, move pass down

parent a8409aba
...@@ -76,7 +76,10 @@ void auto_contiguous::apply(module& m) const ...@@ -76,7 +76,10 @@ void auto_contiguous::apply(module& m) const
if(new_args != args) if(new_args != args)
{ {
m.replace_instruction(ins, ins->get_operator(), new_args); if(not ins->module_inputs().empty())
m.replace_instruction(ins, ins->get_operator(), new_args, ins->module_inputs());
else
m.replace_instruction(ins, ins->get_operator(), new_args);
} }
// shape s = ins->get_shape(); // shape s = ins->get_shape();
// if(not s.dynamic() and not s.standard() and s.elements() != 0) // if(not s.dynamic() and not s.standard() and s.elements() != 0)
......
...@@ -132,7 +132,6 @@ std::vector<pass> target::get_passes(migraphx::context& gctx, const compile_opti ...@@ -132,7 +132,6 @@ std::vector<pass> target::get_passes(migraphx::context& gctx, const compile_opti
dead_code_elimination{}, dead_code_elimination{},
prefuse_ops{}, prefuse_ops{},
dead_code_elimination{}, dead_code_elimination{},
auto_contiguous{},
optimize_module{}, optimize_module{},
fuse_pointwise{}, fuse_pointwise{},
dead_code_elimination{}, dead_code_elimination{},
...@@ -144,6 +143,7 @@ std::vector<pass> target::get_passes(migraphx::context& gctx, const compile_opti ...@@ -144,6 +143,7 @@ std::vector<pass> target::get_passes(migraphx::context& gctx, const compile_opti
dead_code_elimination{}, dead_code_elimination{},
enable_pass(mlir_enabled(), fuse_mlir{&ctx}), enable_pass(mlir_enabled(), fuse_mlir{&ctx}),
dead_code_elimination{}, dead_code_elimination{},
auto_contiguous{},
lowering{&ctx, options.offload_copy}, lowering{&ctx, options.offload_copy},
eliminate_contiguous{"gpu::contiguous"}, eliminate_contiguous{"gpu::contiguous"},
dead_code_elimination{}, dead_code_elimination{},
......
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