Skip to content

Commit 38e57d6

Browse files
committed
Bug fix for intermediate support in Yaml
1 parent 8df9176 commit 38e57d6

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -780,7 +780,7 @@ def GenerateForwardDefinition(self, is_inplaced):
780780
for name, (rtype, pos) in forward_outputs_position_map.items():
781781
if name in intermediate_outputs:
782782
continue
783-
if num_outputs == 1:
783+
if num_outputs == 1 and len(intermediate_outputs) == 0:
784784
returns_list[0] = f"api_result"
785785
else:
786786
# Tuple api_result

python/paddle/utils/code_gen/sparse_api.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
kernel :
55
func : sparse_conv3d
66
layout : x
7-
# intermediate : rulebook
7+
intermediate : rulebook
88
backward : conv3d_grad
99

1010
- api : to_dense

0 commit comments

Comments
 (0)