Bug fix for intermediate support in Yaml (#40935)

PaddlePaddle · Mar 28, 2022 · 3f4099e · 3f4099e
1 parent b64f611
commit 3f4099e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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
@@ -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
@@ -4,7 +4,7 @@
   kernel :
     func : sparse_conv3d
     layout : x
-  # intermediate : rulebook
+  intermediate : rulebook
   backward : conv3d_grad
 
 - api : to_dense