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

Bug fix for intermediate support in Yaml (#40935)

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