"...targets/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "04ae9b80d4b5676911a9dfbb27805d477b1d889f"
Commit 8b81e111 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

changed the eliminate_contiguous pass to skip the reshape operator, so onnx...

changed the eliminate_contiguous pass to skip the reshape operator, so onnx file exported from lstm operator can work correctly in the bidirectional case.
parent 6c70c8a7
...@@ -27,6 +27,13 @@ void eliminate_contiguous::apply(program& p) const ...@@ -27,6 +27,13 @@ void eliminate_contiguous::apply(program& p) const
{ {
for(auto ins : iterator_for(p)) for(auto ins : iterator_for(p))
{ {
// skip the reshape operator for now, since there is a bug
// for the transpose followed by a reshape
if (ins->name() == "reshape")
{
continue;
}
// Make a copy so we can modify it while we iterate // Make a copy so we can modify it while we iterate
auto args = ins->inputs(); auto args = ins->inputs();
for(auto arg : ins->inputs()) for(auto arg : ins->inputs())
......
...@@ -46,8 +46,8 @@ std::vector<pass> target::get_passes(migraphx::context& gctx) const ...@@ -46,8 +46,8 @@ std::vector<pass> target::get_passes(migraphx::context& gctx) const
lowering{ctx}, lowering{ctx},
eliminate_concat{concat_gpu_optimization{}}, eliminate_concat{concat_gpu_optimization{}},
dead_code_elimination{}, dead_code_elimination{},
// eliminate_contiguous{}, eliminate_contiguous{},
// dead_code_elimination{}, dead_code_elimination{},
fuse_ops{&ctx}, fuse_ops{&ctx},
dead_code_elimination{}, dead_code_elimination{},
write_literals{&ctx}, write_literals{&ctx},
......
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