Commit 164959ce authored by Martin Reinecke's avatar Martin Reinecke

more setup.py cleanup

parent ebf44663
...@@ -11,10 +11,9 @@ class _deferred_pybind11_include(object): ...@@ -11,10 +11,9 @@ class _deferred_pybind11_include(object):
return pybind11.get_include(self.user) return pybind11.get_include(self.user)
extra_compile_args = []
include_dirs = ['./', _deferred_pybind11_include(True), include_dirs = ['./', _deferred_pybind11_include(True),
_deferred_pybind11_include()] _deferred_pybind11_include()]
extra_compile_args = []
python_module_link_args = [] python_module_link_args = []
if sys.platform == 'darwin': if sys.platform == 'darwin':
...@@ -24,17 +23,18 @@ if sys.platform == 'darwin': ...@@ -24,17 +23,18 @@ if sys.platform == 'darwin':
vars['LDSHARED'] = vars['LDSHARED'].replace('-bundle', '') vars['LDSHARED'] = vars['LDSHARED'].replace('-bundle', '')
python_module_link_args+=['-bundle'] python_module_link_args+=['-bundle']
else: else:
extra_compile_args += ['--std=c++11', '-march=native', '-O3', '-Wfatal-errors', '-Wno-ignored-attributes', '-DPOCKETFFT_OPENMP', '-fopenmp', '-Wfloat-conversion' ,'-Wsign-conversion', '-Wconversion' ,'-W', '-Wall', '-s'] extra_compile_args += ['--std=c++11', '-march=native', '-O3', '-Wfatal-errors', '-Wno-ignored-attributes', '-DPOCKETFFT_OPENMP', '-fopenmp', '-Wfloat-conversion' ,'-Wsign-conversion', '-Wconversion' ,'-W', '-Wall']
python_module_link_args += ['-march=native', '-Wl,-rpath,$ORIGIN', '-fopenmp'] python_module_link_args += ['-march=native', '-Wl,-rpath,$ORIGIN', '-fopenmp']
# if you don't want debugging info, add "-s" to python_module_link_args
def get_extension_modules(): def get_extension_modules():
pocketfft_library = Extension('pypocketfft', return [Extension('pypocketfft',
sources=['pypocketfft.cc'], sources=['pypocketfft.cc'],
include_dirs=include_dirs, include_dirs=include_dirs,
extra_compile_args=extra_compile_args, extra_compile_args=extra_compile_args,
extra_link_args=python_module_link_args) extra_link_args=python_module_link_args)]
return [pocketfft_library]
setup(name='pypocketfft', setup(name='pypocketfft',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment