[CustomOp] Fix custom op exporting api error. (#59318)

xiaoxiaohehe001 · web-flow · commit 9c74b7b0619b · 2023-11-29T20:14:11.000+08:00
* fix_customop_api

* fix_windows_ci

* fix_lower_in_names
diff --git a/python/paddle/utils/cpp_extension/extension_utils.py b/python/paddle/utils/cpp_extension/extension_utils.py
@@ -1075,18 +1075,28 @@ def _gen_output_content(
 {indent}else:
 {indent}    res.append(None)
 {indent}    start_idx += 1"""
-            static_content += f"""
+            if IS_WINDOWS:
+                static_content += f"""
 {indent}if {lower_in_names} is not None:
 {indent}    outs['{out_name}'] = [helper.create_variable(dtype='float32') for _ in range(len({lower_in_names}))]"""
+            else:
+                static_content += f"""
+{indent}if {lower_in_names} is not None:
+{indent}    outs['{out_name}'] = {lower_in_names}"""
+
         elif (
             in_idx != -1 and "@VECTOR" in in_names[in_idx]
         ):  # inplace vector<Tensor> output case
             lower_in_names = in_names[in_idx].split("@")[0].lower()
             dynamic_content += f"""
 {indent}res.append(outs[start_idx: start_idx + len({lower_in_names})])
 {indent}start_idx += len({lower_in_names})"""
-            static_content += f"""
+            if IS_WINDOWS:
+                static_content += f"""
 {indent}outs['{out_name}'] = [helper.create_variable(dtype='float32') for _ in range(len({lower_in_names}))]"""
+            else:
+                static_content += f"""
+{indent}outs['{out_name}'] = {lower_in_names}"""
         elif (
             in_idx != -1 and "@OPTIONAL" in in_names[in_idx]
         ):  # inplace optional Tensor output case, handle inplace None input
@@ -1097,9 +1107,23 @@ def _gen_output_content(
 {indent}else:
 {indent}    res.append(None)
 {indent}start_idx += 1"""
-            static_content += f"""
+            if IS_WINDOWS:
+                static_content += f"""
 {indent}if {lower_in_names} is not None:
 {indent}    outs['{out_name}'] = helper.create_variable(dtype='float32')"""
+            else:
+                static_content += f"""
+{indent}if {lower_in_names} is not None:
+{indent}    outs['{out_name}'] = {lower_in_names}"""
+        elif (
+            in_idx != -1 and not IS_WINDOWS
+        ):  # inplace Tensor output case, handle inplace None input
+            lower_in_names = in_names[in_idx].lower()
+            dynamic_content += f"""
+{indent}res.append(outs[start_idx])
+{indent}start_idx += 1"""
+            static_content += f"""
+{indent}outs['{out_name}'] = {lower_in_names}"""
         else:  # general/inplace Tensor output case
             dynamic_content += f"""
 {indent}res.append(outs[start_idx])
