From 1492442f99e32ba0fe61ab3402d6284c582834c4 Mon Sep 17 00:00:00 2001
From: kswang <wangkaisheng2@huawei.com>
Date: Tue, 1 Sep 2020 22:49:20 +0800
Subject: [PATCH] set context when init

---
 mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc b/mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc
index d23c1774c9..a58a1c0862 100644
--- 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;