rpm-mpi-hooks/mpi.prov
DistroBaker a6a1a60b39 Merged update from upstream sources
This is an automated DistroBaker update from upstream sources.
If you do not know what this is about or would like to opt out,
contact the OSCI team.

Source: https://src.fedoraproject.org/rpms/rpm-mpi-hooks.git#44dbc1d59e1201d3ead0b70b67aec40e7e4fd036
2021-02-18 15:20:19 +00:00

89 lines
3.2 KiB
Bash
Executable File

#!/bin/bash
# Copyright (c) 2015-2016 Sandro Mani <manisandro@gmail.com>
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE
buildroot="$RPM_BUILD_ROOT"
if [ ! -d "$buildroot" ]; then
>&2 echo "Invalid buildroot"
exit 1
fi
# Arguments are passed to elfdeps
elfdepsargs="$@"
# Search for all MPI implementations:
# - Search all modules which start with mpi
# - For each module, store $MPI_DIR -> $MPI_COMPILER in assoc. array
declare -A mpi_compiler_for_dir
if [ -e /etc/profile.d/modules.sh ]; then
. /etc/profile.d/modules.sh;
# Add moduledirs in buildroot also for cases where an MPI implementation
# package is being built, meaning that its module files are not yet installed
IFS=':' read -ra _MODULEPATH <<< "$MODULEPATH"
for i in "${!_MODULEPATH[@]}"; do
_MODULEPATH[$i]=$buildroot${_MODULEPATH[$i]}
done
export MODULEPATH="$MODULEPATH:$(IFS=:; echo "${_MODULEPATH[*]}")"
for module in $(module -t avail 2>&1 | grep "^mpi/."); do
module load $module
mpi_compiler_for_dir[${MPI_HOME}]=${MPI_COMPILER}
mpi_compiler_for_dir[${MPI_FORTRAN_MOD_DIR}]=${MPI_COMPILER}
if [ -n "${MPI_PYTHON2_SITEARCH}" ]; then
mpi_compiler_for_dir[${MPI_PYTHON2_SITEARCH}]=${MPI_COMPILER}
elif [ -n "${MPI_PYTHON_SITEARCH}" ]; then
mpi_compiler_for_dir[${MPI_PYTHON_SITEARCH}]=${MPI_COMPILER}
fi
if [ -n "${MPI_PYTHON3_SITEARCH}" ]; then
mpi_compiler_for_dir[${MPI_PYTHON3_SITEARCH}]=${MPI_COMPILER}
fi
module unload $module
done
fi
# Read file list from stdin
filelist=$(sed "s/['\"]/\\\&/g")
# List provides:
if [ -x /usr/lib/rpm/elfdeps -a -n "$filelist" ]; then
for file in $(echo $filelist | tr '[:blank:]' \\n); do
# Get the default provides string from elfdeps
prov=$(echo $file | /usr/lib/rpm/elfdeps --provides $elfdepsargs)
if [ -z "$prov" ]; then
continue
fi
# If the path to the scanned binary starts with a known mpi binary directory,
# append the corresponding ($MPI_COMPILER) to the provides string
for mpi_dir in "${!mpi_compiler_for_dir[@]}"; do
if [[ "$file" == "$buildroot$mpi_dir"* ]]; then
prov="${prov}(${mpi_compiler_for_dir[$mpi_dir]})"
break
fi
done
echo "$prov"
done
fi