experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #22734 from mslacken:master

Browse Source 
PiperOrigin-RevId: 219667359
This commit is contained in:
TensorFlower Gardener 2018-11-01 11:29:40 -07:00
commit f7c861f159
1 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
configure.py
View File

@ -1418,11 +1418,16 @@ def set_mpi_home(environ_cp):
def valid_mpi_path(mpi_home): def valid_mpi_path(mpi_home):
exists = ( exists = (
os.path.exists(os.path.join(mpi_home, 'include')) and os.path.exists(os.path.join(mpi_home, 'include')) and
os.path.exists(os.path.join(mpi_home, 'lib'))) (os.path.exists(os.path.join(mpi_home, 'lib')) or
os.path.exists(os.path.join(mpi_home, 'lib64')) or
os.path.exists(os.path.join(mpi_home, 'lib32'))))
if not exists: if not exists:
print('Invalid path to the MPI Toolkit. %s or %s cannot be found' % print(
(os.path.join(mpi_home, 'include'), 'Invalid path to the MPI Toolkit. %s or %s or %s or %s cannot be found'
os.path.exists(os.path.join(mpi_home, 'lib')))) % (os.path.join(mpi_home, 'include'),
os.path.exists(os.path.join(mpi_home, 'lib')),
os.path.exists(os.path.join(mpi_home, 'lib64')),
os.path.exists(os.path.join(mpi_home, 'lib32'))))
return exists return exists
_ = prompt_loop_or_load_from_env( _ = prompt_loop_or_load_from_env(
@ -1463,8 +1468,17 @@ def set_other_mpi_vars(environ_cp):
if os.path.exists(os.path.join(mpi_home, 'lib/libmpi.so')): if os.path.exists(os.path.join(mpi_home, 'lib/libmpi.so')):
symlink_force( symlink_force(
os.path.join(mpi_home, 'lib/libmpi.so'), 'third_party/mpi/libmpi.so') os.path.join(mpi_home, 'lib/libmpi.so'), 'third_party/mpi/libmpi.so')
elif os.path.exists(os.path.join(mpi_home, 'lib64/libmpi.so')):
symlink_force(
os.path.join(mpi_home, 'lib64/libmpi.so'), 'third_party/mpi/libmpi.so')
elif os.path.exists(os.path.join(mpi_home, 'lib32/libmpi.so')):
symlink_force(
os.path.join(mpi_home, 'lib32/libmpi.so'), 'third_party/mpi/libmpi.so')
else: else:
raise ValueError('Cannot find the MPI library file in %s/lib' % mpi_home) raise ValueError(
'Cannot find the MPI library file in %s/lib or %s/lib64 or %s/lib32' %
mpi_home, mpi_home, mpi_home)
def set_system_libs_flag(environ_cp): def set_system_libs_flag(environ_cp):
@ -1681,4 +1695,3 @@ def main():
if __name__ == '__main__': if __name__ == '__main__':
main() main()