Browse Source

!9830 fix cpu pynative coredump

From: @chujinjin
Reviewed-by: @zhoufeng54,@limingqi107
Signed-off-by: @limingqi107
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
f914fa80f9
2 changed files with 8 additions and 1 deletions
  1. +5
    -0
      mindspore/ccsrc/backend/session/session_basic.cc
  2. +3
    -1
      tests/st/pynative/test_pynative_lenet.py

+ 5
- 0
mindspore/ccsrc/backend/session/session_basic.cc View File

@@ -1719,6 +1719,11 @@ void SessionBasic::UpdateGraphDynamicShapeAttr(const NotNull<KernelGraphPtr> &ro
}

void SessionBasic::CleanUselessTensorsImpl(const std::shared_ptr<std::vector<tensor::TensorPtr>> &useless_tensors) {
auto ms_context = MsContext::GetInstance();
std::string device_target = ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET);
if (device_target == "CPU") {
return;
}
for (const auto &tensor : *useless_tensors) {
MS_EXCEPTION_IF_NULL(tensor);
const auto &shape = tensor->shape();


+ 3
- 1
tests/st/pynative/test_pynative_lenet.py View File

@@ -131,9 +131,11 @@ class GradWrap(nn.Cell):
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_cpu
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_ascend_pynative_lenet():
context.set_context(mode=context.PYNATIVE_MODE, device_target="Ascend")
context.set_context(mode=context.PYNATIVE_MODE)

epoch_size = 20
batch_size = 32


Loading…
Cancel
Save