refine cpp extension (#73196)

risemeup1 · web-flow · commit 598fd26be14b · 2025-06-10T11:33:59.000+08:00
diff --git a/python/paddle/utils/cpp_extension/cpp_extension.py b/python/paddle/utils/cpp_extension/cpp_extension.py
@@ -819,6 +819,19 @@ def _record_op_info(self) -> None:
                     op_name, so_name=so_name, so_path=so_path
                 )
 
+    def _clean_intermediate_files(self):
+        for ext in self.extensions:
+            build_dir = os.path.dirname(self.get_ext_fullpath(ext.name))
+            for root, _, files in os.walk(build_dir):
+                for file in files:
+                    if file.endswith(".cu.o") or file.endswith('.o'):
+                        os.remove(os.path.join(root, file))
+                        print(f"Removed: {os.path.join(root, file)}")
+
+    def run(self):
+        super().run()
+        self._clean_intermediate_files()
+
 
 class EasyInstallCommand(easy_install):
     """
