Browse Source

fix bug when MixedPrecisionCast op return a tuple with many elements

tags/v1.2.0-rc1
buxue 5 years ago
parent
commit
985772c192
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      mindspore/ccsrc/pipeline/pynative/pynative_execute.cc

+ 5
- 1
mindspore/ccsrc/pipeline/pynative/pynative_execute.cc View File

@@ -569,7 +569,11 @@ py::object RunOp(const py::args &args) {
py::object PynativeExecutor::RunOpInner(const OpExecInfoPtr &op_exec_info) {
MS_EXCEPTION_IF_NULL(op_exec_info);
if (op_exec_info->op_name == prim::kPrimMixedPrecisionCast->name()) {
return RunOpWithInitBackendPolicy(op_exec_info)[0];
py::tuple ret = RunOpWithInitBackendPolicy(op_exec_info);
if (ret.size() == 1) {
return ret[0];
}
return std::move(ret);
}
// make cnode for building grad graph if grad flag is set.
abstract::AbstractBasePtrList args_spec_list;


Loading…
Cancel
Save