提交 495e73d7 编写于 作者: Y Yancey1989

enable gc

上级 28cdfbc2
...@@ -292,10 +292,11 @@ ParallelExecutor::ParallelExecutor( ...@@ -292,10 +292,11 @@ ParallelExecutor::ParallelExecutor(
graphs.push_back(std::move(graph)); graphs.push_back(std::move(graph));
#endif #endif
auto max_memory_size = GetEagerDeletionThreshold(); auto max_memory_size = GetEagerDeletionThreshold();
// TODO(Yancey1989): fix gc failed on ParallelGraph strategy. if (max_memory_size >= 0) {
if (max_memory_size >= 0 && !build_strategy.enable_parallel_graph_) { for (size_t i = 0; i < graphs.size(); ++i) {
graphs[0] = member_->PrepareGCAndRefCnts( graphs[i] = member_->PrepareGCAndRefCnts(
std::move(graphs[0]), static_cast<size_t>(max_memory_size)); std::move(graphs[i]), static_cast<size_t>(max_memory_size));
}
} }
// Step 3. Create vars in each scope. Passes may also create new vars. // Step 3. Create vars in each scope. Passes may also create new vars.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册