PVD/metrics/ChamferDistancePytorch/chamfer5D/setup.py

18 lines
505 B
Python
Raw Normal View History

2021-10-19 20:54:46 +00:00
from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
setup(
2023-04-11 09:12:58 +00:00
name="chamfer_5D",
2021-10-19 20:54:46 +00:00
ext_modules=[
2023-04-11 09:12:58 +00:00
CUDAExtension(
"chamfer_5D",
[
"/".join(__file__.split("/")[:-1] + ["chamfer_cuda.cpp"]),
"/".join(__file__.split("/")[:-1] + ["chamfer5D.cu"]),
],
),
2021-10-19 20:54:46 +00:00
],
2023-04-11 09:12:58 +00:00
extra_cuda_cflags=["--compiler-bindir=/softs/gcc/11.2.0/bin/gcc"],
cmdclass={"build_ext": BuildExtension},
)