diff --git a/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py b/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py index fd750c0d07369..32aee26e93ccd 100644 --- a/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py +++ b/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py @@ -780,7 +780,7 @@ def GenerateForwardDefinition(self, is_inplaced): 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 diff --git a/python/paddle/utils/code_gen/sparse_api.yaml b/python/paddle/utils/code_gen/sparse_api.yaml index c43e1752152ba..0d0c0c7afbbba 100644 --- a/python/paddle/utils/code_gen/sparse_api.yaml +++ b/python/paddle/utils/code_gen/sparse_api.yaml @@ -4,7 +4,7 @@ kernel : func : sparse_conv3d layout : x - # intermediate : rulebook + intermediate : rulebook backward : conv3d_grad - api : to_dense