nixpkgs/pkgs/development/libraries/science/math/openblas/default.nix

169 lines
4.7 KiB
Nix
Raw Normal View History

{ stdenv, fetchFromGitHub, fetchpatch, gfortran, perl, which, config
# Most packages depending on openblas expect integer width to match
# pointer width, but some expect to use 32-bit integers always
# (for compatibility with reference BLAS).
, blas64 ? null
2019-01-05 12:46:40 +00:00
, buildPackages
# Select a specific optimization target (other than the default)
# See https://github.com/xianyi/OpenBLAS/blob/develop/TargetList.txt
, target ? null
}:
2014-08-28 21:39:01 +00:00
with stdenv.lib;
2015-02-26 12:44:03 +00:00
2015-10-11 14:16:27 +00:00
let blas64_ = blas64; in
let
setTarget = x: if target == null then x else target;
# To add support for a new platform, add an element to this set.
configs = {
2017-12-29 22:19:15 +00:00
armv6l-linux = {
BINARY = 32;
TARGET = setTarget "ARMV6";
DYNAMIC_ARCH = false;
USE_OPENMP = true;
2017-12-29 22:19:15 +00:00
};
armv7l-linux = {
BINARY = 32;
TARGET = setTarget "ARMV7";
DYNAMIC_ARCH = false;
USE_OPENMP = true;
};
2014-08-28 21:39:01 +00:00
2017-12-07 23:59:50 +00:00
aarch64-linux = {
BINARY = 64;
TARGET = setTarget "ARMV8";
DYNAMIC_ARCH = true;
USE_OPENMP = true;
2017-12-07 23:59:50 +00:00
};
i686-linux = {
BINARY = 32;
TARGET = setTarget "P2";
DYNAMIC_ARCH = true;
USE_OPENMP = true;
};
x86_64-darwin = {
BINARY = 64;
TARGET = setTarget "ATHLON";
DYNAMIC_ARCH = true;
USE_OPENMP = false;
MACOSX_DEPLOYMENT_TARGET = "10.7";
};
x86_64-linux = {
BINARY = 64;
TARGET = setTarget "ATHLON";
DYNAMIC_ARCH = true;
USE_OPENMP = true;
};
};
in
let
config =
configs.${stdenv.hostPlatform.system}
or (throw "unsupported system: ${stdenv.hostPlatform.system}");
in
let
blas64 =
if blas64_ != null
then blas64_
else hasPrefix "x86_64" stdenv.hostPlatform.system;
# Convert flag values to format OpenBLAS's build expects.
# `toString` is almost what we need other than bools,
# which we need to map {true -> 1, false -> 0}
# (`toString` produces empty string `""` for false instead of `0`)
mkMakeFlagValue = val:
if !builtins.isBool val then toString val
else if val then "1" else "0";
mkMakeFlagsFromConfig = mapAttrsToList (var: val: "${var}=${mkMakeFlagValue val}");
2015-10-11 14:16:27 +00:00
in
2018-03-12 17:40:49 +00:00
stdenv.mkDerivation rec {
2014-08-28 21:39:01 +00:00
name = "openblas-${version}";
version = "0.3.5";
src = fetchFromGitHub {
owner = "xianyi";
repo = "OpenBLAS";
rev = "v${version}";
sha256 = "0hwfplr6ciqjvfqkya5vz92z2rx8bhdg5mkh923z246ylhs6d94k";
2014-08-28 21:39:01 +00:00
};
2015-10-11 14:16:27 +00:00
inherit blas64;
2016-10-20 21:37:50 +00:00
# Some hardening features are disabled due to sporadic failures in
# OpenBLAS-based programs. The problem may not be with OpenBLAS itself, but
# with how these flags interact with hardening measures used downstream.
# In either case, OpenBLAS must only be used by trusted code--it is
# inherently unsuitable for security-conscious applications--so there should
# be no objection to disabling these hardening measures.
hardeningDisable = [
# don't modify or move the stack
"stackprotector" "pic"
# don't alter index arithmetic
"strictoverflow"
# don't interfere with dynamic target detection
2016-10-20 21:37:50 +00:00
"relro" "bindnow"
];
nativeBuildInputs = [
perl
which
buildPackages.gfortran
buildPackages.stdenv.cc
];
makeFlags = mkMakeFlagsFromConfig (config // {
FC = "${stdenv.cc.targetPrefix}gfortran";
CC = "${stdenv.cc.targetPrefix}${if stdenv.cc.isClang then "clang" else "cc"}";
PREFIX = placeholder "out";
NUM_THREADS = 64;
INTERFACE64 = blas64;
NO_STATIC = true;
CROSS = stdenv.hostPlatform != stdenv.buildPlatform;
HOSTCC = "cc";
2019-02-22 18:46:58 +00:00
# Makefile.system only checks defined status
2019-03-30 16:55:24 +00:00
# This seems to be a bug in the openblas Makefile:
# on x86_64 it expects NO_BINARY_MODE=
# but on aarch64 it expects NO_BINARY_MODE=0
NO_BINARY_MODE = if stdenv.isx86_64
then toString (stdenv.hostPlatform != stdenv.buildPlatform)
else stdenv.hostPlatform != stdenv.buildPlatform;
});
doCheck = true;
2015-10-11 14:16:27 +00:00
checkTarget = "tests";
2018-08-17 12:49:52 +00:00
postInstall = ''
# Write pkgconfig aliases. Upstream report:
# https://github.com/xianyi/OpenBLAS/issues/1740
for alias in blas cblas lapack; do
cat <<EOF > $out/lib/pkgconfig/$alias.pc
2018-08-17 12:49:52 +00:00
Name: $alias
Version: ${version}
Description: $alias provided by the OpenBLAS package.
Cflags: -I$out/include
Libs: -L$out/lib -lopenblas
EOF
done
'';
2014-08-29 15:02:39 +00:00
meta = with stdenv.lib; {
2014-08-28 21:39:01 +00:00
description = "Basic Linear Algebra Subprograms";
2014-08-29 15:02:39 +00:00
license = licenses.bsd3;
homepage = https://github.com/xianyi/OpenBLAS;
platforms = platforms.unix;
2014-08-29 15:02:39 +00:00
maintainers = with maintainers; [ ttuegel ];
2014-08-28 21:39:01 +00:00
};
2018-08-14 06:00:54 +00:00
# We use linkName to pass a different name to --with-blas-libs for
# fflas-ffpack and linbox, because we use blas on darwin but openblas
# elsewhere.
# See see https://github.com/NixOS/nixpkgs/pull/45013.
passthru.linkName = "openblas";
2014-08-28 21:39:01 +00:00
}