|
|
|
@ -148,13 +148,13 @@ void SetKernelInfo(const CNodePtr &kernel_node) {
|
|
|
|
|
std::vector<size_t> input_not_cnode_indexes;
|
|
|
|
|
std::vector<std::string> output_formats;
|
|
|
|
|
std::vector<TypeId> output_types;
|
|
|
|
|
|
|
|
|
|
MS_LOG(INFO) << "SetKernelInfo, CNode Name: " << AnfAlgo::GetCNodeName(kernel_node);
|
|
|
|
|
GetInputFormatsAndDtypes(kernel_node, &input_formats, &input_types, &input_not_cnode_indexes);
|
|
|
|
|
|
|
|
|
|
auto kernel_attrs =
|
|
|
|
|
kernel::CPUKernelFactory::GetInstance().GetSupportedKernelAttrList(AnfAlgo::GetCNodeName(kernel_node));
|
|
|
|
|
|
|
|
|
|
if (kernel_attrs.empty()) {
|
|
|
|
|
MS_LOG(EXCEPTION) << "Operator[" << AnfAlgo::GetCNodeName(kernel_node) << "] is not support.";
|
|
|
|
|
}
|
|
|
|
|
int max_type_matched_num = -1;
|
|
|
|
|
int max_format_matched_num = -1;
|
|
|
|
|
KernelAttr selected_kernel_attr;
|
|
|
|
|