From 036ddef0df37cd2a95b38680366dd10fd2cf9e6f Mon Sep 17 00:00:00 2001 From: Cristian C Lalescu <Cristian.Lalescu@ds.mpg.de> Date: Fri, 31 Jul 2015 15:03:19 +0200 Subject: [PATCH] move local settings to outside file --- machine_settings.py | 33 +++++++++++++++++++++++++++++++++ setup.py | 16 +++------------- 2 files changed, 36 insertions(+), 13 deletions(-) create mode 100644 machine_settings.py diff --git a/machine_settings.py b/machine_settings.py new file mode 100644 index 00000000..a4f7d29b --- /dev/null +++ b/machine_settings.py @@ -0,0 +1,33 @@ +import os + +######################################################################## +#### these you're supposed to adapt to your environment +######################################################################## + +hostname = os.getenv('HOSTNAME') + +extra_compile_args = ['-mtune=native', '-ffast-math', '-std=c++11'] + +include_dirs = [] +library_dirs = [] + +if hostname in ['frontend01', 'frontend02']: + include_dirs = ['/usr/lib64/mpi/gcc/openmpi/include', + '/usr/include/mpich', + '/usr/nld/mvapich2-1.9a2-gcc/include', + '/usr/nld/fftw-3.3.3-mvapich2-1.9a2-gcc/include', + '/usr/nld/fftw-3.3.3-float-mvapich2-1.9a2-gcc/include'] + + library_dirs = ['/usr/nld/fftw-3.3.3-mvapich2-1.9a2-gcc/lib', + '/usr/nld/fftw-3.3.3-float-mvapich2-1.9a2-gcc/lib'] + +if hostname == 'tolima': + local_install_dir = '/scratch.local/chichi/installs' + + include_dirs = ['/usr/lib64/mpi/gcc/openmpi/include', + '/usr/include/mpich', + os.path.join(local_install_dir, 'include')] + + library_dirs = [os.path.join(local_install_dir, 'lib'), + os.path.join(local_install_dir, 'lib64')] + diff --git a/setup.py b/setup.py index fbffcdd4..e5aa603f 100644 --- a/setup.py +++ b/setup.py @@ -22,16 +22,7 @@ import os -######################################################################## -#### these you're supposed to adapt to your environment - -# first off, this is the install folder for fftw and/or openmpi etc -local_install_dir = '/scratch.local/chichi/installs' - -include_dirs = ['/usr/lib64/mpi/gcc/openmpi/include', - '/usr/include/mpich', - os.path.join(local_install_dir, 'include')] -######################################################################## +from machine_settings import include_dirs, library_dirs, extra_compile_args AUTHOR = 'Cristian C Lalescu' @@ -72,9 +63,8 @@ libbfps = Extension( sources = ['bfps/cpp/' + fname + '.cpp' for fname in src_file_list], include_dirs = include_dirs, libraries = libraries, - extra_compile_args = ['-mtune=native', '-ffast-math', '-std=c++11'], - library_dirs = [os.path.join(local_install_dir, 'lib'), - os.path.join(local_install_dir, 'lib64')]) + extra_compile_args = extra_compile_args, + library_dirs = library_dirs) setup( name = 'bfps', -- GitLab