Ejemplo n.º 1
0
EXTS.append(
    Extension('selectinf.randomized.selective_MLE_utils', [
        'selectinf/randomized/selective_MLE_utils.pyx',
        'C-software/src/selective_mle.c', 'C-software/src/cox_fns.c'
    ],
              libraries=[],
              include_dirs=['C-software/src']))

# Cython is a dependency for building extensions, iff we don't have stamped
# up pyx and c files.
build_ext, need_cython = cyproc_exts(EXTS, info.CYTHON_MIN_VERSION,
                                     'pyx-stamps')

# Add numpy includes when building extension.
build_ext = make_np_ext_builder(build_ext)

# Check dependencies, maybe add to setuptools lists
if need_cython:
    SetupDependency('Cython',
                    info.CYTHON_MIN_VERSION,
                    req_type='install_requires',
                    heavy=False).check_fill(extra_setuptools_args)
SetupDependency('numpy',
                info.NUMPY_MIN_VERSION,
                req_type='install_requires',
                heavy=True).check_fill(extra_setuptools_args)
SetupDependency('scipy',
                info.SCIPY_MIN_VERSION,
                req_type='install_requires',
                heavy=True).check_fill(extra_setuptools_args)
Ejemplo n.º 2
0
build_ext, need_cython = cyproc_exts(EXTS,
                                     info.CYTHON_MIN_VERSION,
                                     'pyx-stamps')
# Add openmp flags if they work
simple_test_c = """int main(int argc, char** argv) { return(0); }"""
omp_test_c = """#include <omp.h>
int main(int argc, char** argv) { return(0); }"""
extbuilder = add_flag_checking(
    build_ext, [
        [['/arch:SSE2'], [], simple_test_c, 'USING_VC_SSE2'],
        [['-msse2', '-mfpmath=sse'], [], simple_test_c, 'USING_GCC_SSE2'],
        [['-fopenmp'], ['-fopenmp'], omp_test_c, 'HAVE_OPENMP']], 'dipy')

# Use ext builder to add np.get_include() at build time, not during setup.py
# execution.
extbuilder = make_np_ext_builder(extbuilder)
if need_cython:
    SetupDependency('Cython', info.CYTHON_MIN_VERSION,
                    req_type='install_requires',
                    heavy=True).check_fill(extra_setuptools_args)
SetupDependency('numpy', info.NUMPY_MIN_VERSION,
                req_type='install_requires',
                heavy=True).check_fill(extra_setuptools_args)
SetupDependency('scipy', info.SCIPY_MIN_VERSION,
                req_type='install_requires',
                heavy=True).check_fill(extra_setuptools_args)
SetupDependency('nibabel', info.NIBABEL_MIN_VERSION,
                req_type='install_requires',
                heavy=False).check_fill(extra_setuptools_args)

cmdclass = dict(
Ejemplo n.º 3
0
# Test if it is a 32 bits version
if not sys.maxsize > 2**32:
    # This flag is needed only on 32 bits
    msc_flag_defines += [
        [['/arch:SSE2'], [], simple_test_c, 'USING_VC_SSE2'],
    ]

flag_defines = msc_flag_defines if 'msc' in platform.python_compiler().lower(
) else gcc_flag_defines

extbuilder = add_flag_checking(build_ext, flag_defines, 'dipy')

# Use ext builder to add np.get_include() at build time, not during setup.py
# execution.
extbuilder = make_np_ext_builder(extbuilder)
if need_cython:
    SetupDependency('Cython',
                    info.CYTHON_MIN_VERSION,
                    req_type='install_requires',
                    heavy=True).check_fill(extra_setuptools_args)
SetupDependency('numpy',
                info.NUMPY_MIN_VERSION,
                req_type='install_requires',
                heavy=True).check_fill(extra_setuptools_args)
SetupDependency('scipy',
                info.SCIPY_MIN_VERSION,
                req_type='install_requires',
                heavy=True).check_fill(extra_setuptools_args)
SetupDependency('nibabel',
                info.NIBABEL_MIN_VERSION,
Ejemplo n.º 4
0
EXTS = []
for modulename, other_sources in (
    ("regreg.atoms.projl1_cython", []),
    ("regreg.atoms.mixed_lasso_cython", []),
    ("regreg.atoms.piecewise_linear", []),
):
    pyx_src = pjoin(*modulename.split(".")) + ".pyx"
    EXTS.append(Extension(modulename, [pyx_src] + other_sources))


# Cython is a dependency for building extensions, iff we don't have stamped
# up pyx and c files.
build_ext, need_cython = cyproc_exts(EXTS, info.CYTHON_MIN_VERSION, "pyx-stamps")

# Add numpy includes when building extension.
build_ext = make_np_ext_builder(build_ext)

# Check dependencies, maybe add to setuptools lists
if need_cython:
    SetupDependency("Cython", info.CYTHON_MIN_VERSION, req_type="install_requires", heavy=False).check_fill(
        extra_setuptools_args
    )
SetupDependency("numpy", info.NUMPY_MIN_VERSION, req_type="install_requires", heavy=True).check_fill(
    extra_setuptools_args
)
SetupDependency("scipy", info.SCIPY_MIN_VERSION, req_type="install_requires", heavy=True).check_fill(
    extra_setuptools_args
)


cmdclass = versioneer.get_cmdclass()