!1985 GPU update datatype info for akg kernel

Merge pull request !1985 from VectorSL/fix-akg-type-info
pull/1985/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 259341b9ba

@ -82,11 +82,16 @@ std::string SupportedTypeList(const CNodePtr &kernel_node) {
(void)ParseMetadata(kernel_node, op_info_ptr, kernel::Processor::CUDA, &kernel_info_list);
for (size_t i = 0; i < kernel_info_list.size(); i++) {
auto supported_akg_type = kernel_info_list[i]->GetAllInputDeviceTypes();
std::string supported_akg_type_list = "[";
auto supported_akg_type_out = kernel_info_list[i]->GetAllOutputDeviceTypes();
std::string supported_akg_type_list = "in[";
for (auto type : supported_akg_type) {
supported_akg_type_list = supported_akg_type_list + mindspore::kernel::TypeId2String(type);
}
supported_type_lists = supported_type_lists + supported_akg_type_list + "] ";
supported_type_lists = supported_type_lists + supported_akg_type_list + "], out[";
for (auto type : supported_akg_type_out) {
supported_akg_type_list = supported_akg_type_list + mindspore::kernel::TypeId2String(type);
}
supported_type_lists += "]; ";
}
return supported_type_lists;
}

Loading…
Cancel
Save