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