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

!579 fix hccl get data type bug 

Merge pull request !579 from Maoweiyong/fix_hccl_allreduce_input_check
This commit is contained in:
mindspore-ci-bot 2020-04-22 20:41:37 +08:00 committed by Gitee
parent 140201fcd2 b50bfbf7d0
commit 4e85ca68e8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/kernel/hccl/hcom_util.cc
View File

@ -49,7 +49,7 @@ bool HcomUtil::GetHcomDataType(const AnfNodePtr &anf_node, vector<hcclDataType_t
MS_EXCEPTION_IF_NULL(anf_node);
MS_EXCEPTION_IF_NULL(data_type_list);
for (size_t i = 0; i < AnfAlgo::GetInputTensorNum(anf_node); ++i) {
auto type_ptr = AnfAlgo::GetPrevNodeOutputInferDataType(anf_node, i);
auto type_ptr = AnfAlgo::GetPrevNodeOutputDeviceDataType(anf_node, i);
auto iter = CONST_OP_HCOM_DATA_TYPE_MAP.find(type_ptr);
if (iter == CONST_OP_HCOM_DATA_TYPE_MAP.end()) {
MS_LOG(EXCEPTION) << "HcomDataType cann't support Current Ascend Data Type : " << type_ptr;