!706 subgraph optimize multi thread env add

* subgraph optimize multi thread env
pull/706/MERGE
王笑天 4 years ago
parent cfbdb031c8
commit b5ccf9c834

@ -526,7 +526,15 @@ Status GraphManager::OptimizeSubGraphWithMultiThreads(ComputeGraphPtr compute_gr
Graph2SubGraphInfoList &sub_graph_map, uint64_t session_id) {
GE_CHECK_NOTNULL(compute_graph);
// use default 16 multi thread
const uint32_t thread_num = 16;
uint32_t thread_num = 16;
char *env = std::getenv("THREAD_MULTI_NUM");
if (env != nullptr) {
thread_num = atoi(env);
GEEVENT("OptimizeSubGraphWithMultiThreads thread num: %u", thread_num);
}
ThreadPool executor(thread_num);
std::vector<std::future<Status>> vector_future;
const auto &root_subgraph_list = sub_graph_map[compute_graph];

Loading…
Cancel
Save