PVD/metrics/ChamferDistancePytorch/chamfer2D/setup.py
2023-04-11 11:12:58 +02:00

18 lines
505 B
Python

from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
setup(
name="chamfer_2D",
ext_modules=[
CUDAExtension(
"chamfer_2D",
[
"/".join(__file__.split("/")[:-1] + ["chamfer_cuda.cpp"]),
"/".join(__file__.split("/")[:-1] + ["chamfer2D.cu"]),
],
),
],
extra_cuda_cflags=["--compiler-bindir=/softs/gcc/11.2.0/bin/gcc"],
cmdclass={"build_ext": BuildExtension},
)