LION/third_party/ChamferDistancePytorch/chamfer6D/setup.py

14 lines
398 B
Python
Raw Normal View History

2023-01-23 05:14:49 +00:00
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
})