Browse Source

!732 om md5 consistent

Merge pull request !732 from yangyongqiang/om_md5_same
pull/734/head
yangyongqiang i-robot 2 years ago
parent
commit
1c550e0a90
1 changed files with 13 additions and 1 deletions
  1. +13
    -1
      parser/common/thread_pool.cc

+ 13
- 1
parser/common/thread_pool.cc View File

@@ -21,10 +21,22 @@
#include <stdexcept>

#include "register/register_types.h"
#include "graph/ge_context.h"
#include "mmpa/mmpa_api.h"

namespace ge {
namespace {
const char* const kMultiThreadCompile = "MULTI_THREAD_COMPILE";
const char* const kDisEnableFlag = "0";
bool IsSingleThreadCompile() {
std::string compile_thread;
return ((ge::GetContext().GetOption(kMultiThreadCompile, compile_thread) == GRAPH_SUCCESS)
&& (compile_thread.compare(kDisEnableFlag) == 0));
}
}

FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY ThreadPool::ThreadPool(uint32_t size) : is_stoped_(false) {
idle_thrd_num_ = size < 1 ? 1 : size;
idle_thrd_num_ = ((size < 1U) || IsSingleThreadCompile()) ? 1U : size;

for (uint32_t i = 0; i < idle_thrd_num_; ++i) {
pool_.emplace_back(ThreadFunc, this);


Loading…
Cancel
Save