mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!3031 fix context mode and device target dependency issue 

Merge pull request !3031 from wangqiuliang/fix-tensor-bool-operation
This commit is contained in:
mindspore-ci-bot 2020-07-13 19:52:32 +08:00 committed by Gitee
parent 8947afb5bf 8863dfd677
commit c465ea3269
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/context.py
View File

@ -176,10 +176,7 @@ class _Context:
self._context_switches.push(True, None) self._context_switches.push(True, None)
else: else:
if self.enable_debug_runtime: if self.enable_debug_runtime:
if self.device_target == "CPU": self.set_backend_policy("ge")
self.set_backend_policy("vm")
else:
self.set_backend_policy("ge")
self._context_switches.push(False, None) self._context_switches.push(False, None)
def set_backend_policy(self, policy): def set_backend_policy(self, policy):
@ -221,6 +218,8 @@ class _Context:
success = self._context_handle.set_device_target(target) success = self._context_handle.set_device_target(target)
if not success: if not success:
raise ValueError("Target device name is invalid!!!") raise ValueError("Target device name is invalid!!!")
if self.enable_debug_runtime and self.device_target == "CPU":
self.set_backend_policy("vm")
@property @property
def device_id(self): def device_id(self):

3
tests/ut/python/pipeline/parse/test_cell_bprop.py
View File

@ -29,8 +29,7 @@ from .....mindspore_test_framework.utils.bprop_util import bprop
def setup_module(module): def setup_module(module):
context.set_context(device_target="CPU") context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
context.set_context(mode=context.GRAPH_MODE)
def teardown_module(module): def teardown_module(module):
context.set_context(device_target="Ascend") context.set_context(device_target="Ascend")