KPConv-PyTorch/cpp_wrappers/cpp_neighbors/setup.py
Laurent FAINSIN d0cdb8e4ee 🎨 black + ruff
2023-05-15 17:18:10 +02:00

27 lines
546 B
Python

from distutils.core import setup, Extension
import numpy.distutils.misc_util
# Adding OpenCV to project
# ************************
# Adding sources of the project
# *****************************
SOURCES = [
"../cpp_utils/cloud/cloud.cpp",
"neighbors/neighbors.cpp",
"wrapper.cpp",
]
module = Extension(
name="radius_neighbors",
sources=SOURCES,
extra_compile_args=["-std=c++11", "-D_GLIBCXX_USE_CXX11_ABI=0"],
)
setup(
ext_modules=[module],
include_dirs=numpy.distutils.misc_util.get_numpy_include_dirs(),
)