fix log print

pull/655/head
liubuyu 5 years ago
parent 84d5e4f923
commit 80e5bb982b

@ -543,11 +543,6 @@ bool IsValidKernelInfo(const std::shared_ptr<CNode> &kernel_node, const kernel::
if (!IsShapeMatchFormat(shape, format)) { if (!IsShapeMatchFormat(shape, format)) {
return false; return false;
} }
for (auto shape_value : shape) {
if (shape_value == 0) {
MS_LOG(EXCEPTION) << "Dimension size of the tensor shape should be a positive integer, but got " << shape_value;
}
}
return true; return true;
}; };
for (size_t index = 0; index < kernel_build_info.GetOutputNum(); ++index) { for (size_t index = 0; index < kernel_build_info.GetOutputNum(); ++index) {
@ -593,10 +588,12 @@ void TbeMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<Ke
if (context_ptr->execution_mode() == kPynativeMode) { if (context_ptr->execution_mode() == kPynativeMode) {
kernel_info_list->push_back(parse_info); kernel_info_list->push_back(parse_info);
} else { } else {
if (IsValidKernelInfo(kernel_node, *(parse_info)) && CheckSupported(kernel_node, parse_info)) { if (IsValidKernelInfo(kernel_node, *(parse_info))) {
kernel_info_list->push_back(parse_info); if (CheckSupported(kernel_node, parse_info)) {
} else { kernel_info_list->push_back(parse_info);
MS_LOG(INFO) << "CheckSupported Failed for TBE op" << op_name << " kernel info."; } else {
MS_LOG(INFO) << "CheckSupported Failed for TBE op" << op_name << " kernel info.";
}
} }
} }
} }

Loading…
Cancel
Save