transforms84/setup.py at main · Stoops-ML/transforms84 · 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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
import os
import platform
import sys

import numpy as np
from setuptools import Extension, setup

include_dirs = [
    np.get_include(),
    os.path.join(os.path.dirname(os.path.realpath(__file__)), "include"),
]

if sys.platform.startswith("win"):
    if "MSC" in sys.version:
        ompcompileflags = ["-openmp"]
        omplinkflags = []
    else:
        # For non-MSVC toolchain e.g. gcc and clang with mingw
        ompcompileflags = ["-fopenmp"]
        omplinkflags = ["-fopenmp"]
elif sys.platform.startswith("darwin"):
    if "clang" in os.popen("cc --version").read():
        # Disable OpenMP for macOS with clang
        ompcompileflags = []
        omplinkflags = []
    else:
        # llvm (clang) OpenMP is used for headers etc at compile time
        # Intel OpenMP (libiomp5) provides the link library.
        ompcompileflags = ["-fopenmp"]
        omplinkflags = ["-fopenmp=libiomp5"]
        omppath = ["lib", "clang", "*", "include", "omp.h"]
        include_dirs.append("/".join(omppath))  # Add omppath to include_dirs
else:
    ompcompileflags = ["-fopenmp"]
    if platform.machine() == "ppc64le":  # from Numba # noqa: SIM108
        omplinkflags = ["-fopenmp"]
    else:
        omplinkflags = ["-fopenmp"]

setup(
    ext_modules=[
        Extension(
            "transforms84.transforms",
            sources=["include/transforms.c"],
            include_dirs=include_dirs,
            extra_compile_args=ompcompileflags,
            extra_link_args=omplinkflags,
        ),
        Extension(
            "transforms84.distances",
            sources=["include/distances.c"],
            include_dirs=include_dirs,
            extra_compile_args=ompcompileflags,
            extra_link_args=omplinkflags,
        ),
        Extension(
            "transforms84.helpers",
            sources=["include/helpers.c"],
            include_dirs=include_dirs,
            extra_compile_args=ompcompileflags,
            extra_link_args=omplinkflags,
        ),
    ],
)