Browse Source

!1985 GPU update datatype info for akg kernel

Merge pull request !1985 from VectorSL/fix-akg-type-info
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
259341b9ba
1 changed files with 7 additions and 2 deletions
  1. +7
    -2
      mindspore/ccsrc/device/gpu/kernel_info_setter.cc

+ 7
- 2
mindspore/ccsrc/device/gpu/kernel_info_setter.cc View File

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


Loading…
Cancel
Save