diff --git a/lite/core/mir/memory_optimize_pass.cc b/lite/core/mir/memory_optimize_pass.cc index b46b933afebea53675d4f6482aa006d5e277675b..5bf018eb872662681afbbf37936a34b55bb60ace 100644 --- a/lite/core/mir/memory_optimize_pass.cc +++ b/lite/core/mir/memory_optimize_pass.cc @@ -52,21 +52,21 @@ void MemoryOptimizePass::CollectLifeCycleByDevice( "feed", "fetch"}; - auto insert_invalid_op_nodes_for_specific_target = - [&](std::set op_node_set, TargetType specific_target) { - std::set invalid_op_nodes_opencl = {"layout", "fc"}; - for (auto& op_node : graph->StmtTopologicalOrder()) { - if (!op_node->IsStmt()) continue; - TargetType op_target_type = op_node->AsStmt().place().target; - if (op_target_type == specific_target && - specific_target == TARGET(kOpenCL)) { - invalid_op_nodes.insert(invalid_op_nodes_opencl.begin(), - invalid_op_nodes_opencl.end()); - break; - } - // else if // you can add more targets - } - }; + auto insert_invalid_op_nodes_for_specific_target = [&]( + std::set op_node_set, TargetType specific_target) { + std::set invalid_op_nodes_opencl = {"layout", "fc"}; + for (auto& op_node : graph->StmtTopologicalOrder()) { + if (!op_node->IsStmt()) continue; + TargetType op_target_type = op_node->AsStmt().place().target; + if (op_target_type == specific_target && + specific_target == TARGET(kOpenCL)) { + invalid_op_nodes.insert(invalid_op_nodes_opencl.begin(), + invalid_op_nodes_opencl.end()); + break; + } + // else if // you can add more targets + } + }; VLOG(4) << "invalid_op_nodes.size();" << invalid_op_nodes.size(); insert_invalid_op_nodes_for_specific_target(invalid_op_nodes,