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 f23582bdd1588..fa0a29525658e 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 @@ -860,7 +860,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 770aaeae88453..293fdc1528a12 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