From d4ed0aab5671f83e9338f62f0ef03d940fdb6492 Mon Sep 17 00:00:00 2001 From: Guillaume Horel Date: Thu, 23 Aug 2018 13:10:00 -0400 Subject: use openmp to compute pv --- setup.py | 30 ++++++++++++++++-------------- 1 file changed, 16 insertions(+), 14 deletions(-) (limited to 'setup.py') diff --git a/setup.py b/setup.py index 2254fe0..359540d 100644 --- a/setup.py +++ b/setup.py @@ -4,22 +4,24 @@ from Cython.Build import cythonize import numpy all_extensions = Extension("*", ["pyisda/*.pyx"], - include_dirs = ['c_layer', numpy.get_include()], - libraries = ["cds", "farmhash"], - language = 'c++') + include_dirs=['c_layer', numpy.get_include()], + libraries=["cds", "farmhash"], + language='c++', + extra_compile_args=['-fopenmp'], + extra_link_args=['-fopenmp', '-Wl,--strip-all']) c_extension = Extension("pyisda.flat_hazard", - include_dirs = ['c_layer', numpy.get_include()], - sources = ['pyisda/flat_hazard.pyx', 'c_layer/cdsbootstrap.c'], - libraries = ['cds'], - language = 'c++') + include_dirs=['c_layer', numpy.get_include()], + sources=['pyisda/flat_hazard.pyx', 'c_layer/cdsbootstrap.c'], + libraries=['cds'], + language='c++') -all_extensions = cythonize([c_extension, all_extensions], nthreads = 4, - compiler_directives={'embedsignature':True}) +all_extensions = cythonize([c_extension, all_extensions], nthreads=4, + compiler_directives={'embedsignature': True}) setup( - name = "pyisda", - version = '0.1', - author = 'Guillaume Horel', - ext_modules = all_extensions, - packages = ['pyisda']) + name="pyisda", + version='0.1', + author='Guillaume Horel', + ext_modules=all_extensions, + packages=['pyisda']) -- cgit v1.2.3-70-g09d2