LION/third_party/ChamferDistancePytorch/chamfer5D/setup.py
2023-01-23 00:14:49 -05:00

14 lines
398 B
Python
Executable file

from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
setup(
name='chamfer_5D',
ext_modules=[
CUDAExtension('chamfer_5D', [
"/".join(__file__.split('/')[:-1] + ['chamfer_cuda.cpp']),
"/".join(__file__.split('/')[:-1] + ['chamfer5D.cu']),
]),
],
cmdclass={
'build_ext': BuildExtension
})