diff --git a/ge/graph/manager/graph_manager.cc b/ge/graph/manager/graph_manager.cc index 6a13e285..117b243f 100755 --- a/ge/graph/manager/graph_manager.cc +++ b/ge/graph/manager/graph_manager.cc @@ -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> vector_future; const auto &root_subgraph_list = sub_graph_map[compute_graph];