From dc8a279eb6d6d620616bff419bb440e863840b1f Mon Sep 17 00:00:00 2001 From: simson Date: Thu, 18 Mar 2021 17:04:12 +0800 Subject: [PATCH] add fp16 for const input -> valuenode --- mindspore/ccsrc/pipeline/pynative/pynative_execute.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc index be58574030..abdbc36588 100644 --- a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc +++ b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc @@ -294,7 +294,8 @@ std::string GetSingleOpGraphInfo(const OpExecInfoPtr &op_exec_info, if (input_tensors[index]->Dtype()->type_id() == kNumberTypeInt64) { (void)graph_info.append(std::to_string(*reinterpret_cast(input_tensors[index]->data_c()))); graph_info += "_"; - } else if (input_tensors[index]->Dtype()->type_id() == kNumberTypeFloat32) { + } else if (input_tensors[index]->Dtype()->type_id() == kNumberTypeFloat32 || + input_tensors[index]->Dtype()->type_id() == kNumberTypeFloat16) { (void)graph_info.append(std::to_string(*reinterpret_cast(input_tensors[index]->data_c()))); graph_info += "_"; } else {