!742 GPU update kernel type verification

Merge pull request !742 from VectorSL/update-type-check
pull/742/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 14191c4ba9

@ -68,13 +68,18 @@ std::string GpuKernelFactory::SupportedTypeList(const std::string &kernel_name)
return type_lists;
}
for (size_t attr_index = 0; attr_index < (iter->second).size(); ++attr_index) {
std::string type_list = "[";
std::string type_list = "in[";
auto attr = (iter->second)[attr_index].first;
for (size_t input_index = 0; input_index < attr.GetInputSize(); ++input_index) {
type_list = type_list + TypeId2String(attr.GetInputAttr(input_index).first) +
((input_index == (attr.GetInputSize() - 1)) ? "" : " ");
}
type_lists = type_lists + type_list + "] ";
type_list = type_list + "], out[";
for (size_t input_index = 0; input_index < attr.GetOutputSize(); ++input_index) {
type_list = type_list + TypeId2String(attr.GetOutputAttr(input_index).first) +
((input_index == (attr.GetOutputSize() - 1)) ? "" : " ");
}
type_lists = type_lists + type_list + "]; ";
}
return type_lists;
}

Loading…
Cancel
Save