KPConv-PyTorch/cpp_wrappers/cpp_neighbors/setup.py

27 lines
546 B
Python
Raw Normal View History

2020-03-31 19:42:35 +00:00
from distutils.core import setup, Extension
import numpy.distutils.misc_util
# Adding OpenCV to project
# ************************
# Adding sources of the project
# *****************************
2023-05-15 15:18:10 +00:00
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(),
)