setup.py

from setuptools import setup
from torch.utils import cpp_extension
import sys
try:
    from setuptools import setup
except ImportError:
    from distutils.core import setup
from distutils.extension import Extension
from Cython.Build import cythonize
from torch.utils.cpp_extension import BuildExtension, CppExtension, CUDAExtension
import numpy

# print(sys.argv)

setup(
    name="lightconvpoint",
    ext_modules=[],
    cmdclass={}
)

# Get the numpy include directory.
numpy_include_dir = numpy.get_include()



# triangle hash (efficient mesh intersection)
triangle_hash_module = Extension(
    'eval.src.utils.libmesh.triangle_hash',
    sources=[
        'eval/src/utils/libmesh/triangle_hash.pyx'
    ],
    libraries=['m'],  # Unix-like specific
    include_dirs=[numpy_include_dir]
)


# Gather all extension modules
ext_modules = [
    triangle_hash_module,
]

setup(
    ext_modules=cythonize(ext_modules),
    cmdclass={
        'build_ext': BuildExtension
    }
)