Commit 44369c8e authored by Umang Yadav's avatar Umang Yadav
Browse files

refactor

parent d493b4ca
...@@ -149,14 +149,15 @@ struct auto_gen_root_modules ...@@ -149,14 +149,15 @@ struct auto_gen_root_modules
} }
} }
bool is_different_subgraph(migraphx::instruction_ref ins, bool is_different_subgraph(migraphx::instruction_ref current_ins,
std::optional<std::size_t> previous_tid) std::optional<std::size_t> previous_tid)
{ {
if(tass.find(ins) == tass.end()) if(tass.find(current_ins) == tass.end())
{ {
return previous_tid.has_value(); return previous_tid.has_value();
} }
return tass.at(ins) != previous_tid.value_or(std::numeric_limits<std::size_t>::max()); return tass.at(current_ins) !=
previous_tid.value_or(std::numeric_limits<std::size_t>::max());
} }
/* /*
...@@ -178,15 +179,7 @@ struct auto_gen_root_modules ...@@ -178,15 +179,7 @@ struct auto_gen_root_modules
return false; return false;
} }
return std::any_of(inputs.begin(), inputs.end(), [&](auto input_ins) { return std::any_of(inputs.begin(), inputs.end(), [&](auto input_ins) {
if(tass.find(input_ins) == tass.end()) return is_different_subgraph(input_ins, ins_tid);
{
return ins_tid.has_value();
}
else
{
return tass.at(input_ins) !=
ins_tid.value_or(std::numeric_limits<std::size_t>::max());
}
}); });
} }
...@@ -208,15 +201,7 @@ struct auto_gen_root_modules ...@@ -208,15 +201,7 @@ struct auto_gen_root_modules
return false; return false;
} }
return std::any_of(outputs.begin(), outputs.end(), [&](auto output_ins) { return std::any_of(outputs.begin(), outputs.end(), [&](auto output_ins) {
if(tass.find(output_ins) == tass.end()) return is_different_subgraph(output_ins, ins_tid);
{
return ins_tid.has_value();
}
else
{
return tass.at(output_ins) !=
ins_tid.value_or(std::numeric_limits<std::size_t>::max());
};
}); });
} }
......
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