Rev | Line | |
---|
[6320] | 1 | from distutils.core import setup |
---|
| 2 | from distutils.extension import Extension |
---|
| 3 | from Cython.Distutils import build_ext |
---|
| 4 | import numpy as np |
---|
| 5 | import os |
---|
| 6 | |
---|
| 7 | os.environ["CC"] = "gcc-4.8" |
---|
| 8 | |
---|
| 9 | # Run the following from this directory: |
---|
| 10 | # python setup.py build_ext --inplace |
---|
| 11 | |
---|
| 12 | ext_module = Extension( |
---|
| 13 | "coll_util_fast", |
---|
| 14 | ["coll_util_fast.pyx"], |
---|
| 15 | extra_compile_args=['-fopenmp'], |
---|
| 16 | extra_link_args=['-fopenmp'], |
---|
| 17 | ) |
---|
| 18 | |
---|
| 19 | setup( |
---|
| 20 | name = 'Collision Utility (Fast)', |
---|
| 21 | cmdclass = {'build_ext': build_ext}, |
---|
| 22 | include_dirs = [np.get_include()], |
---|
| 23 | ext_modules = [ext_module], |
---|
| 24 | ) |
---|
Note: See
TracBrowser
for help on using the repository browser.