Browse Source

fix async

pull/328/head
chuxing 5 years ago
parent
commit
146ec8f71b
3 changed files with 3 additions and 3 deletions
  1. +1
    -1
      ge/hybrid/executor/hybrid_model_async_executor.cc
  2. +1
    -1
      ge/hybrid/executor/rt_callback_manager.cc
  3. +1
    -1
      ge/hybrid/executor/subgraph_executor.cc

+ 1
- 1
ge/hybrid/executor/hybrid_model_async_executor.cc View File

@@ -58,7 +58,7 @@ Status HybridModelAsyncExecutor::Start(const std::shared_ptr<ModelListener> &lis

run_flag_ = true;
listener_ = listener;
future_ = std::async([&]() -> Status {
future_ = std::async(std::launch::async, [&]() -> Status {
GetContext().SetSessionId(executor_->GetContext()->session_id);
return RunInternal();
});


+ 1
- 1
ge/hybrid/executor/rt_callback_manager.cc View File

@@ -49,7 +49,7 @@ Status CallbackManager::RegisterCallback(rtCallback_t callback, void *user_data)
Status CallbackManager::Init() {
rtContext_t ctx = nullptr;
GE_CHK_RT_RET(rtCtxGetCurrent(&ctx));
ret_future_ = std::async([&](rtContext_t context) ->Status {
ret_future_ = std::async(std::launch::async, [&](rtContext_t context) ->Status {
return CallbackProcess(context);
}, ctx);
if (!ret_future_.valid()) {


+ 1
- 1
ge/hybrid/executor/subgraph_executor.cc View File

@@ -307,7 +307,7 @@ Status SubgraphExecutor::LaunchTasks() {

Status SubgraphExecutor::ScheduleTasks() {
GELOGD("[%s] Start to schedule prepare workers.", graph_item_->GetName().c_str());
auto prepare_future = std::async([&]() -> Status {
auto prepare_future = std::async(std::launch::async, [&]() -> Status {
GetContext().SetSessionId(context_->session_id);
auto ret = PrepareNodes();
ready_queue_.Push(nullptr);


Loading…
Cancel
Save