From b902f485a66f199830a16afb5299ebb99011b08c Mon Sep 17 00:00:00 2001 From: dengwentao Date: Mon, 20 Apr 2020 16:10:23 +0800 Subject: [PATCH] check tbe attr required --- mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc b/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc index 496f99df1c..d2ad014e5b 100644 --- a/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc +++ b/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc @@ -383,6 +383,10 @@ bool TbeKernelJsonCreator::GenTbeAttrJson(const std::shared_ptr &anf_no attr_obj["name"] = attr_name; attr_obj["valid"] = true; (*attrs_json).push_back(attr_obj); + } else { + if (attr_ptr->param_type() == "required" && creater_type_ == SINGLE_BUILD && op_info->impl_path() != "") { + MS_LOG(EXCEPTION) << "op name: " << op_info->op_name() << " attr: " << attr_name << "is required, but not set."; + } } } return true;