!558 add verification for tbe op attr

Merge pull request !558 from dengwentao/tbe_attr
pull/558/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 2cdb62629f

@ -383,6 +383,10 @@ bool TbeKernelJsonCreator::GenTbeAttrJson(const std::shared_ptr<AnfNode> &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;

Loading…
Cancel
Save