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