diff --git a/machine_settings.py b/machine_settings.py new file mode 100644 index 0000000000000000000000000000000000000000..a4f7d29b6f43b36cceb1fded4a91338a37af0e04 --- /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 fbffcdd47c42711120ea3e15f787b4de695d82e3..e5aa603f73286351b4c4079ce67edbebfaeae384 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',