gsaligner/setup.py at main · rvp-group/gsaligner · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
from setuptools import setup
from torch.utils.cpp_extension import CUDAExtension, BuildExtension, CppExtension
import os

os.path.dirname(os.path.abspath(__file__))

BASE_DIR = os.path.dirname(os.path.abspath(__file__))

setup(
    name="gsaligner",
    packages=["gsaligner"],
    version="0.0.1",
    ext_modules=[
        CUDAExtension(
            name="gsaligner._C",
            sources=[
                "gsaligner/aligner.cu",
                "gsaligner/aligner_utils.cu",
                "ext.cpp",
                "gsaligner/tools/image_set.cpp",
                "gsaligner/tools/image_set_impl.cu",
                "gsaligner/tools/mad_tree.cpp",
                "gsaligner/tools/sum_reduce.cu",
                "gsaligner/registration/geometric.cpp",
                "gsaligner/registration/photometric.cpp",
                "gsaligner/registration/photometric_cuda.cu",
                "gsaligner/registration/vel_estimator.cpp",
            ],
            include_dirs=[
                os.path.join(BASE_DIR, "third_party/eigen"),
                os.path.join(BASE_DIR, "gsaligner"),
                os.path.join(BASE_DIR, "gsaligner/tools"),
                os.path.join(BASE_DIR, "gsaligner/registration"),
            ],
        )
    ],
    cmdclass={"build_ext": BuildExtension},
)