python-numpy-c-extension-examples/setup.py at master · johnnylee/python-numpy-c-extension-examples · 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
#!/usr/bin/env python

from distutils.core import setup, Extension

setup(name             = "numpy_c_ext_example",
      version          = "1.0",
      description      = "Example code for blog post.",
      author           = "J. David Lee",
      author_email     = "contact@crumpington.com",
      maintainer       = "contact@crumpington.com",
      url              = "https://www.crumpington.com",
      ext_modules      = [
          Extension(
              'lib.simple1', ['src/simple1.c'],
              extra_compile_args=["-Ofast", "-march=native"]),
          Extension(
              'lib.simple2', ['src/simple2.c'],
              extra_compile_args=["-Ofast", "-march=native"]),
          Extension(
              'lib.simd1', ['src/simd1.c'],
              extra_compile_args=["-Ofast", "-march=native"]),
          Extension(
              'lib.omp1', ['src/omp1.c'],
              extra_compile_args=["-Ofast", "-march=native", "-fopenmp"],
              libraries=["gomp"]),
          Extension(
              'lib.simdomp1', ['src/simdomp1.c'],
              extra_compile_args=["-Ofast", "-march=native", "-fopenmp"],
              libraries=["gomp"]),
      ],

)