modify setup.py to fix issues related to 3d ops (#3417)

setup.py

@@ -50,6 +50,21 @@ def version():
         return file.read().rstrip()
 
 
+def get_data_files(directory: str, filetypes: list = None):
+    all_files = []
+    filetypes = filetypes or []
+
+    for root, _, files in os.walk(directory):
+        rel_root = os.path.relpath(root, directory)
+        for file in files:
+            filepath = os.path.join(rel_root, file)
+            filetype = os.path.splitext(file)[1][1:]
+            if filetype in filetypes:
+                all_files.append(filepath)
+
+    return all_files
+
+
 def packages_and_package_data():
     """get packages and package_data"""
 
@@ -83,7 +98,11 @@ def packages_and_package_data():
     pkg_data.append("serving_requirements.txt")
     pkg_data.append("paddle2onnx_requirements.txt")
     pkg_data.append("hpip_links.html")
-    return pkgs, {"paddlex": pkg_data}
+    ops_file_dir = 'paddlex/ops'
+    ops_file_types = ['h', 'hpp', 'cpp', 'cc', 'cu']
+    return pkgs, {
+        "paddlex.ops": get_data_files(ops_file_dir, ops_file_types),
+        "paddlex": pkg_data}
 
 
 if __name__ == "__main__":