diff --git a/mindspore/ccsrc/device/gpu/kernel_info_setter.cc b/mindspore/ccsrc/device/gpu/kernel_info_setter.cc index 2ba154b87b..05d6679f76 100644 --- a/mindspore/ccsrc/device/gpu/kernel_info_setter.cc +++ b/mindspore/ccsrc/device/gpu/kernel_info_setter.cc @@ -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; }