Commit 5315d9bb authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

fix bug in eliminate_contiguous

parent c9a533d4
......@@ -76,8 +76,6 @@ void auto_contiguous::apply(module& m) const
{
continue;
}
if(ins->name() == "pointwise")
std::cout << "HERE" << std::endl;
auto c = m.insert_instruction(std::next(ins), make_op("contiguous"), ins);
m.replace_instruction(ins, c);
}
......
......@@ -79,7 +79,7 @@ static bool try_compute_shape(instruction_ref ins,
return (arg == ins) ? new_shape : arg->get_shape();
});
if(not try_compute_shape(output, input_shapes, mods))
if(not try_compute_shape(output, input_shapes, output->module_inputs()))
{
return false;
}
......
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