提交 cf73a54f 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!655 fix log printing problems

Merge pull request !655 from liubuyu/master
...@@ -543,11 +543,6 @@ bool IsValidKernelInfo(const std::shared_ptr<CNode> &kernel_node, const kernel:: ...@@ -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 ...@@ -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.";
}
} }
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册