未验证 提交 3f4099ee 编写于 作者: Z Zhanlue Yang 提交者: GitHub

Bug fix for intermediate support in Yaml (#40935)

上级 b64f611e
......@@ -860,7 +860,7 @@ class DygraphSingleFunctionGenerator(FunctionGeneratorBase):
for name, (rtype, pos) in forward_outputs_position_map.items():
if name in intermediate_outputs:
continue
if num_outputs == 1:
if num_outputs == 1 and len(intermediate_outputs) == 0:
returns_list[0] = f"api_result"
else:
# Tuple api_result
......
......@@ -4,7 +4,7 @@
kernel :
func : sparse_conv3d
layout : x
# intermediate : rulebook
intermediate : rulebook
backward : conv3d_grad
- api : to_dense
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册