init

hy3dpaint/custom_rasterizer/setup.py

@@ -0,0 +1,40 @@
+# Hunyuan 3D is licensed under the TENCENT HUNYUAN NON-COMMERCIAL LICENSE AGREEMENT
+# except for the third-party components listed below.
+# Hunyuan 3D does not impose any additional limitations beyond what is outlined
+# in the repsective licenses of these third-party components.
+# Users must comply with all terms and conditions of original licenses of these third-party
+# components and must ensure that the usage of the third party components adheres to
+# all relevant laws and regulations.
+
+# For avoidance of doubts, Hunyuan 3D means the large language models and
+# their software and algorithms, including trained model weights, parameters (including
+# optimizer states), machine-learning model code, inference-enabling code, training-enabling code,
+# fine-tuning enabling code and other elements of the foregoing made publicly available
+# by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
+
+from setuptools import setup, find_packages
+import torch
+from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CppExtension
+
+# build custom rasterizer
+
+custom_rasterizer_module = CUDAExtension(
+    "custom_rasterizer_kernel",
+    [
+        "lib/custom_rasterizer_kernel/rasterizer.cpp",
+        "lib/custom_rasterizer_kernel/grid_neighbor.cpp",
+        "lib/custom_rasterizer_kernel/rasterizer_gpu.cu",
+    ],
+)
+
+setup(
+    packages=find_packages(),
+    version="0.1",
+    name="custom_rasterizer",
+    include_package_data=True,
+    package_dir={"": "."},
+    ext_modules=[
+        custom_rasterizer_module,
+    ],
+    cmdclass={"build_ext": BuildExtension},
+)