|
|
from setuptools import setup, find_packages |
|
|
from torch.utils.cpp_extension import BuildExtension, CUDAExtension |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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}, |
|
|
) |
|
|
|
