mirror of
https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git
synced 2025-12-10 05:14:29 +08:00
35 lines
1.0 KiB
Python
Executable File
35 lines
1.0 KiB
Python
Executable File
from setuptools import setup, find_packages
|
|
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
|
|
import torch
|
|
|
|
torch_version = torch.__version__.split('+')[0].replace('.', '')
|
|
if torch.version.cuda is not None:
|
|
cuda_version = "cuda"+torch.version.cuda.replace('.', '')
|
|
elif getattr(torch.version, 'hip', None) is not None:
|
|
cuda_version = "rocm" + torch.version.hip.replace('.','')
|
|
|
|
version = f"0.1.0+torch{torch_version}.{cuda_version}"
|
|
# build custom rasterizer
|
|
# build with `python setup.py install`
|
|
# nvcc is needed
|
|
|
|
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=version,
|
|
name='custom_rasterizer',
|
|
include_package_data=True,
|
|
package_dir={'': '.'},
|
|
ext_modules=[
|
|
custom_rasterizer_module,
|
|
],
|
|
cmdclass={
|
|
'build_ext': BuildExtension
|
|
},
|
|
)
|