41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
# 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},
|
|
)
|