14 lines
398 B
Python
14 lines
398 B
Python
|
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
|
||
|
|
})
|