Skip to content

Commit 3f4099e

Browse files
authored
Bug fix for intermediate support in Yaml (#40935)
1 parent b64f611 commit 3f4099e

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
@@ -860,7 +860,7 @@ def GenerateForwardDefinition(self, is_inplaced):
860860
for name, (rtype, pos) in forward_outputs_position_map.items():
861861
if name in intermediate_outputs:
862862
continue
863-
if num_outputs == 1:
863+
if num_outputs == 1 and len(intermediate_outputs) == 0:
864864
returns_list[0] = f"api_result"
865865
else:
866866
# 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)