setup.py

import os
import platform
import glob
from distutils.core import setup
from Cython.Build import cythonize
from setuptools import Extension


def set_gcc():
    """
    Try to find and use GCC on OSX for OpenMP support, setting the CC
    environment variable accordingly.
    """
    # For macports and homebrew
    patterns = ['/opt/local/bin/gcc-mp-[0-9].[0-9]',
                '/opt/local/bin/gcc-mp-[0-9]',
                '/usr/local/bin/gcc-[0-9].[0-9]',
                '/usr/local/bin/gcc-[0-9]']

    gcc_binaries = []
    for pattern in patterns:
        gcc_binaries += glob.glob(pattern)
    gcc_binaries.sort()

    if gcc_binaries:
        _, gcc = os.path.split(gcc_binaries[-1])
        os.environ["CC"] = gcc

    else:
        raise Exception('No GCC available. Install gcc from Homebrew '
                        'using brew install gcc.')

if 'darwin' in platform.platform().lower():
    set_gcc()

args = {'extra_compile_args': ['-fopenmp', '-ffast-math', '-O3'],
        'extra_link_args': ['-fopenmp']}

extensions = [Extension('prng', ['prng.pyx'], **args),
              Extension('fm', ['fm.pyx'], **args)]
setup(ext_modules=cythonize(extensions))