p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!13544 add fp16 for const input -> valuenode 

From: @simson_wu
Reviewed-by: 
Signed-off-by:
This commit is contained in:
mindspore-ci-bot 2021-03-19 13:56:57 +08:00 committed by Gitee
parent 6c096bff31 dc8a279eb6
commit 94841e4889
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
View File

@ -294,7 +294,8 @@ std::string GetSingleOpGraphInfo(const OpExecInfoPtr &op_exec_info,
if (input_tensors[index]->Dtype()->type_id() == kNumberTypeInt64) { if (input_tensors[index]->Dtype()->type_id() == kNumberTypeInt64) {
(void)graph_info.append(std::to_string(*reinterpret_cast<int *>(input_tensors[index]->data_c()))); (void)graph_info.append(std::to_string(*reinterpret_cast<int *>(input_tensors[index]->data_c())));
graph_info += "_"; 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<float *>(input_tensors[index]->data_c()))); (void)graph_info.append(std::to_string(*reinterpret_cast<float *>(input_tensors[index]->data_c())));
graph_info += "_"; graph_info += "_";
} else { } else {