c4d5135fde
The openmp derivation provides both libomp and libompv. To avoid accidentally linking with the wrong library and to avoid the nosv dependency on libomp, this patch separates each version in a different derivation. Also, it adapts the clang wrappers and stdenvs to provide an stdenv per openmp library where each openmp will be used by default when the compiler flag "-fopenmp" is used. This eases linking ompv with nixpkgs libraries, such as blis, that expect openmp to be provided with stdenv. Reviewed-by: Rodrigo Arias Mallo <rodrigo.arias@bsc.es> Tested-by: Rodrigo Arias Mallo <rodrigo.arias@bsc.es>
68 lines
2.4 KiB
Nix
68 lines
2.4 KiB
Nix
{
|
|
stdenv
|
|
, lib
|
|
, gcc
|
|
, clangOmpss2Unwrapped
|
|
, openmp ? null
|
|
, wrapCCWith
|
|
, llvmPackages_latest
|
|
, ompss2rt ? null
|
|
}:
|
|
|
|
let
|
|
usingNodesAndOmpv = (openmp.pname == "openmp-v" && ompss2rt.pname == "nodes");
|
|
sameNosv = openmp.nosv == ompss2rt.nosv;
|
|
in
|
|
|
|
assert lib.assertMsg (usingNodesAndOmpv -> sameNosv) "OpenMP-V and NODES must share the same nOS-V";
|
|
|
|
let
|
|
homevar = if ompss2rt.pname == "nanos6" then "NANOS6_HOME" else "NODES_HOME";
|
|
rtname = if ompss2rt.pname == "nanos6" then "libnanos6" else "libnodes";
|
|
ompname = if openmp.pname == "openmp-v" then "libompv" else "libomp";
|
|
|
|
|
|
# We need to replace the lld linker from bintools with our linker just built,
|
|
# otherwise we run into incompatibility issues when mixing compiler and linker
|
|
# versions.
|
|
bintools-unwrapped = llvmPackages_latest.tools.bintools-unwrapped.override {
|
|
lld = clangOmpss2Unwrapped;
|
|
};
|
|
bintools = llvmPackages_latest.tools.bintools.override {
|
|
bintools = bintools-unwrapped;
|
|
};
|
|
targetConfig = stdenv.targetPlatform.config;
|
|
inherit gcc;
|
|
cc = clangOmpss2Unwrapped;
|
|
in wrapCCWith {
|
|
inherit cc bintools;
|
|
# extraPackages adds packages to depsTargetTargetPropagated
|
|
extraPackages = lib.optional (openmp != null) openmp;
|
|
extraBuildCommands = ''
|
|
echo "-target ${targetConfig}" >> $out/nix-support/cc-cflags
|
|
echo "-B${gcc.cc}/lib/gcc/${targetConfig}/${gcc.version}" >> $out/nix-support/cc-cflags
|
|
echo "-L${gcc.cc}/lib/gcc/${targetConfig}/${gcc.version}" >> $out/nix-support/cc-ldflags
|
|
echo "-L${gcc.cc.lib}/lib" >> $out/nix-support/cc-ldflags
|
|
|
|
for dir in ${gcc.cc}/include/c++/*; do
|
|
echo "-isystem $dir" >> $out/nix-support/libcxx-cxxflags
|
|
done
|
|
for dir in ${gcc.cc}/include/c++/*/${targetConfig}; do
|
|
echo "-isystem $dir" >> $out/nix-support/libcxx-cxxflags
|
|
done
|
|
|
|
echo "--gcc-toolchain=${gcc}" >> $out/nix-support/cc-cflags
|
|
|
|
wrap clang++ $wrapper $ccPath/clang++
|
|
|
|
'' + lib.optionalString (openmp != null) ''
|
|
echo "export OPENMP_RUNTIME=${ompname}" >> $out/nix-support/cc-wrapper-hook
|
|
'' + lib.optionalString (ompss2rt != null) ''
|
|
echo "export OMPSS2_RUNTIME=${rtname}" >> $out/nix-support/cc-wrapper-hook
|
|
echo "export ${homevar}=${ompss2rt}" >> $out/nix-support/cc-wrapper-hook
|
|
'' + lib.optionalString (ompss2rt != null && ompss2rt.pname == "nodes") ''
|
|
echo "export NOSV_HOME=${ompss2rt.nosv}" >> $out/nix-support/cc-wrapper-hook
|
|
'';
|
|
}
|
|
|