未验证 提交 9c5e4b4e 编写于 作者: Z zyfncg 提交者: GitHub

[Yaml] Fix bug of code-gen for op_maker (#55369)

* add check of input tensors in Yaml

* fix bug of code-gen for opmaker

* fix bug
上级 0cea7b7d
......@@ -29,7 +29,7 @@ TODO: Documentation of {{op_name}} op.
{% macro add_input(i, input, op_name) %}{# inline #}
{% set name = input["fluid_name"] %}
{% set typename = input["typename"] %}
AddInput({{name| to_opmaker_name}}, "({{typename}}), input {{i}} of {{op_name}} op.")
AddInput("{{name}}", "({{typename}}), input {{i}} of {{op_name}} op.")
{%- if typename is vec %}
.AsDuplicable()
......@@ -49,7 +49,7 @@ AddInput({{name| to_opmaker_name}}, "({{typename}}), input {{i}} of {{op_name}}
{% set name = output["fluid_name"] %}
{% set typename = output["typename"] %}
{% set is_intermediate = output["intermediate"] %}
AddOutput({{name | to_opmaker_name}}, "({{typename}}), output {{i}} of {{op_name}} op.")
AddOutput("{{name}}", "({{typename}}), output {{i}} of {{op_name}} op.")
{%- if typename is vec %}
.AsDuplicable()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册