Commit edc90b37 authored by Diego Caballero's avatar Diego Caballero Committed by Scott Cyphers

Replace missing set with vector container (#3230)

Fix a mismatch after changing get_inputs API.
parent 579e44b8
...@@ -239,7 +239,7 @@ bool MLIRSubgraphExtractionPass::run_on_function(std::shared_ptr<Function> func) ...@@ -239,7 +239,7 @@ bool MLIRSubgraphExtractionPass::run_on_function(std::shared_ptr<Function> func)
auto& out_desc = output_descs[0]; auto& out_desc = output_descs[0];
// 'replace_output' invalidates iterator of the original container. Use a copy instead. // 'replace_output' invalidates iterator of the original container. Use a copy instead.
const std::set<descriptor::Input*> input_descs = out_desc.get_inputs(); const std::vector<descriptor::Input*> input_descs = out_desc.get_inputs();
for (descriptor::Input* in_desc : input_descs) for (descriptor::Input* in_desc : input_descs)
{ {
......
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