Browse Source

fix cpu pynative mixed precision error

tags/v1.1.0
chujinjin 5 years ago
parent
commit
02c8052e89
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/backend/session/cpu_session.cc

+ 1
- 0
mindspore/ccsrc/backend/session/cpu_session.cc View File

@@ -163,6 +163,7 @@ void CPUSession::SetOutputFlags(const VectorRef &base_ref, std::vector<tensor::T
} else if (utils::isa<tensor::TensorPtr>(base_ref[i])) {
auto tensor_ptr = utils::cast<std::shared_ptr<tensor::Tensor>>(base_ref[i]);
tensor_ptr->SetNeedWait(false);
tensor_ptr->data_sync(false);
outputs_tensors->push_back(tensor_ptr);
}
}


Loading…
Cancel
Save