!5644 set context for worker thread

Merge pull request !5644 from kisnwang/async-run-graph
2020-09-01 23:55:51 +08:00 · 2020-09-01 23:55:51 +08:00 · e5c5708c2c
parent 749979e7c4 1492442f99
commit e5c5708c2c
1 changed files with 1 additions and 0 deletions
--- a/mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc
+++ b/mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc
@ -194,6 +194,7 @@ void AscendKernelRuntime::ReleaseDeviceRes() {

 bool AscendKernelRuntime::Init() {
  if (initialized_) {
+    SetContext();
    return true;
  }
  bool ret = false;