16 lines
470 B
Python
16 lines
470 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']),
|
|
]),
|
|
],
|
|
|
|
extra_cuda_cflags=['--compiler-bindir=/softs/gcc/11.2.0/bin/gcc'],
|
|
cmdclass={
|
|
'build_ext': BuildExtension
|
|
}) |