From eb671300b5b28834ff75282d973257d296dbf4dd Mon Sep 17 00:00:00 2001 From: VectorSL Date: Thu, 11 Jun 2020 09:41:30 +0800 Subject: [PATCH] gpu update datatype info for akg kernel --- mindspore/ccsrc/device/gpu/kernel_info_setter.cc | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) 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; }