products/env_scripts/openmpi.py

   1 #!/usr/bin/env python
   2 #-*- coding:utf-8 -*-
   3
   4 import os.path
   5
   6 def set_env(env, prereq_dir, version):
   7     env.set('OPENMPIDIR', prereq_dir)
   8     env.set('OPAL_PREFIX', prereq_dir) # be able to move openmpi install (packages)
   9     env.set('MPI_ROOT_DIR', prereq_dir)  # update for cmake
  10     env.set('MPI_ROOT', prereq_dir)
  11     root = env.get('OPENMPIDIR')
  12
  13     env.prepend('PATH', os.path.join(root, 'bin'))
  14     env.prepend('PATH', os.path.join(root, 'include'))
  15     env.prepend('LD_LIBRARY_PATH', os.path.join(root, 'lib'))
  16     env.prepend('C_INCLUDE_PATH', os.path.join(root, 'include'))  # needed for parallel h5py
  17
  18 def set_nativ_env(env):
  19     env.set('MPI_ROOT_DIR', "/usr")  # update for cmake
  20     env.set('OPENMPIDIR', "/usr")
  21     env.set('MPI_ROOT', "/usr")
  22
  23