spns #30249: add optional MPI_Fortran_COMPILER

[tools/sat_salome.git] / products / env_scripts / openmpi.py

diff --git a/products/env_scripts/openmpi.py b/products/env_scripts/openmpi.py
index 11e2dc559904a464aa9d061944d61e83bfa47680..7eb0d67b06519c7898b285779e3f19452f907bfd 100644 (file)
--- a/products/env_scripts/openmpi.py
+++ b/products/env_scripts/openmpi.py
@@ -11,6 +11,8 @@ def set_env(env, prereq_dir, version):
     env.set('MPI_INCLUDE_DIR', os.path.join(prereq_dir, 'include'))
     env.set('MPI_C_COMPILER', os.path.join(prereq_dir, 'bin', 'mpicc'))
     env.set('MPI_CXX_COMPILER', os.path.join(prereq_dir, 'bin', 'mpicxx'))
+    env.set('MPI_Fortran_COMPILER', os.path.join(prereq_dir, 'bin', 'mpifort'))
+    env.set('MPIEXEC_EXECUTABLE', os.path.join(prereq_dir, 'bin', 'mpiexec'))
     env.set('MPI_C_FOUND', os.path.join(prereq_dir,'lib','libmpi.so'))
     env.prepend('PATH', os.path.join(prereq_dir, 'bin'))
     env.prepend('PATH', os.path.join(prereq_dir, 'include'))
@@ -23,7 +25,7 @@ def set_nativ_env(env):
     prereq_inc='/usr/include/openmpi'
     try:
         import distro
-        if any(distribution in distro.name().lower() for distribution in ["centos", "fedora"]) :
+        if any(distribution in distro.name().lower() for distribution in ["rocky", "centos", "fedora"]) :
             prereq_dir='/usr/lib64/openmpi'
             prereq_bin='/usr/lib64/openmpi/bin'
             prereq_inc='/usr/include/openmpi-x86_64'
@@ -32,7 +34,7 @@ def set_nativ_env(env):
             prereq_inc= '/usr/lib/x86_64-linux-gnu/openmpi/include'
     except:
         import platform
-        if any(distribution in platform.linux_distribution()[0].lower() for distribution in ["centos", "fedora"]) :
+        if any(distribution in platform.linux_distribution()[0].lower() for distribution in ["rocky", "centos", "fedora"]) :
             prereq_dir='/usr/lib64/openmpi'
             prereq_bin='/usr/lib64/openmpi/bin'
             prereq_inc='/usr/include/openmpi-x86_64'