未验证 提交 0f925968 编写于 作者: Z Zhanlue Yang 提交者: GitHub

Fixed GradNode default attributes issues (#40132)

* Fixed GradNode default attributes issues

* Reverted changes on yaml files
上级 5496a7ab
...@@ -509,11 +509,18 @@ def GenerateNodeDeclaration(fwd_api_name, backward_fwd_input_map, ...@@ -509,11 +509,18 @@ def GenerateNodeDeclaration(fwd_api_name, backward_fwd_input_map,
set_attribute_methods_str += SET_ATTR_METHOD_TEMPLATE.format( set_attribute_methods_str += SET_ATTR_METHOD_TEMPLATE.format(
aname, GetConstReference(atype), aname, saved_attr_name, aname) aname, GetConstReference(atype), aname, saved_attr_name, aname)
ATTRIBUTE_MEMBER_TEMPLATE = """ if default_val:
{} {} = {}; ATTRIBUTE_MEMBER_TEMPLATE = """
""" {} {} = {};
attribute_members_str += ATTRIBUTE_MEMBER_TEMPLATE.format( """
RemoveConstAndReference(atype), saved_attr_name, default_val) attribute_members_str += ATTRIBUTE_MEMBER_TEMPLATE.format(
RemoveConstAndReference(atype), saved_attr_name, default_val)
else:
ATTRIBUTE_MEMBER_TEMPLATE = """
{} {};
"""
attribute_members_str += ATTRIBUTE_MEMBER_TEMPLATE.format(
RemoveConstAndReference(atype), saved_attr_name)
# End: SetAttributes & Attribute Members # End: SetAttributes & Attribute Members
grad_node_name = GetGradNodeName(fwd_api_name) grad_node_name = GetGradNodeName(fwd_api_name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册