Browse Source

fix context mode and device_target dependency issue

tags/v0.6.0-beta
kingfo 5 years ago
parent
commit
8863dfd677
2 changed files with 4 additions and 6 deletions
  1. +3
    -4
      mindspore/context.py
  2. +1
    -2
      tests/ut/python/pipeline/parse/test_cell_bprop.py

+ 3
- 4
mindspore/context.py View File

@@ -176,10 +176,7 @@ class _Context:
self._context_switches.push(True, None)
else:
if self.enable_debug_runtime:
if self.device_target == "CPU":
self.set_backend_policy("vm")
else:
self.set_backend_policy("ge")
self.set_backend_policy("ge")
self._context_switches.push(False, None)

def set_backend_policy(self, policy):
@@ -221,6 +218,8 @@ class _Context:
success = self._context_handle.set_device_target(target)
if not success:
raise ValueError("Target device name is invalid!!!")
if self.enable_debug_runtime and self.device_target == "CPU":
self.set_backend_policy("vm")

@property
def device_id(self):


+ 1
- 2
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):
context.set_context(device_target="CPU")
context.set_context(mode=context.GRAPH_MODE)
context.set_context(mode=context.GRAPH_MODE, device_target="CPU")

def teardown_module(module):
context.set_context(device_target="Ascend")


Loading…
Cancel
Save