|
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
|
|
}
|
|
)
|
|
|
