提交 4e85ca68 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!579 fix hccl get data type bug

Merge pull request !579 from Maoweiyong/fix_hccl_allreduce_input_check
......@@ -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;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册