未验证 提交 e699f81d 编写于 作者: Z zhangbo9674 提交者: GitHub

fix bug (#55200)

上级 d6e259bb
......@@ -97,8 +97,8 @@ OUTPUT_VECTORTYPE_CHECK_TEMPLATE = """
phi::errors::PreconditionNotMet("Type validation failed for the {index}th output."));
}}"""
OUTPUT_OPTIONAL_TYPE_CHECK_TEMPLATE = """
if (auto output_{index} = (*this)->result({index})) {{
PADDLE_ENFORCE(output_{index}.type().isa<{standard}>(),
if (auto output_{index}_type = (*this)->result({index}).type()) {{
PADDLE_ENFORCE(output_{index}_type.isa<{standard}>(),
phi::errors::PreconditionNotMet("Type validation failed for the {index}th output."));
}}"""
OUTPUT_OPTIONAL_VECTORTYPE_CHECK_TEMPLATE = """
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册