gDel3D-python/setup.py at master · blancocd/gDel3D-python · 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
39
40
41
from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension

src_root = 'GDelFlipping/src'

sources = [
    'binding.cu',
    f'{src_root}/gDel3D/GpuDelaunay.cu',
    f'{src_root}/gDel3D/GPU/ThrustWrapper.cu',
    f'{src_root}/gDel3D/GPU/KerPredicates.cu',
    f'{src_root}/gDel3D/GPU/KerDivision.cu',

    # CPU Sources
    f'{src_root}/gDel3D/CPU/predicates.cpp',
    f'{src_root}/RandGen.cpp',

    # Renamed/CUDA sources
    f'{src_root}/gDel3D/CPU/PredWrapper.cu',
    f'{src_root}/gDel3D/CPU/Splaying.cu',
    f'{src_root}/gDel3D/CPU/Star.cu',
    f'{src_root}/InputCreator.cu',
    f'{src_root}/DelaunayChecker.cu'
]

setup(
    name='gdel3d',
    ext_modules=[
        CUDAExtension(
            name='gdel3d',
            sources=sources,
            include_dirs=[src_root, f'{src_root}/gDel3D'],
            extra_compile_args={
                'cxx': ['-O3', '-D_GLIBCXX_USE_CXX11_ABI=1'],
                'nvcc': ['-O3', '-D_GLIBCXX_USE_CXX11_ABI=1']
            }
        )
    ],
    cmdclass={
        'build_ext': BuildExtension
    }
)