提交 d6159419 编写于 作者: J Jiansong Wang

Use clang-foramt 3.8 insteady of 6.0.0, otherwise clang-format will

format memory_optimized_pass.cc again. test=develop
上级 30084365
...@@ -52,21 +52,21 @@ void MemoryOptimizePass::CollectLifeCycleByDevice( ...@@ -52,21 +52,21 @@ void MemoryOptimizePass::CollectLifeCycleByDevice(
"feed", "feed",
"fetch"}; "fetch"};
auto insert_invalid_op_nodes_for_specific_target = auto insert_invalid_op_nodes_for_specific_target = [&](
[&](std::set<std::string> op_node_set, TargetType specific_target) { std::set<std::string> op_node_set, TargetType specific_target) {
std::set<std::string> invalid_op_nodes_opencl = {"layout", "fc"}; std::set<std::string> invalid_op_nodes_opencl = {"layout", "fc"};
for (auto& op_node : graph->StmtTopologicalOrder()) { for (auto& op_node : graph->StmtTopologicalOrder()) {
if (!op_node->IsStmt()) continue; if (!op_node->IsStmt()) continue;
TargetType op_target_type = op_node->AsStmt().place().target; TargetType op_target_type = op_node->AsStmt().place().target;
if (op_target_type == specific_target && if (op_target_type == specific_target &&
specific_target == TARGET(kOpenCL)) { specific_target == TARGET(kOpenCL)) {
invalid_op_nodes.insert(invalid_op_nodes_opencl.begin(), invalid_op_nodes.insert(invalid_op_nodes_opencl.begin(),
invalid_op_nodes_opencl.end()); invalid_op_nodes_opencl.end());
break; break;
} }
// else if // you can add more targets // else if // you can add more targets
} }
}; };
VLOG(4) << "invalid_op_nodes.size();" << invalid_op_nodes.size(); VLOG(4) << "invalid_op_nodes.size();" << invalid_op_nodes.size();
insert_invalid_op_nodes_for_specific_target(invalid_op_nodes, insert_invalid_op_nodes_for_specific_target(invalid_op_nodes,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册