amd-blis: init at 2.2
We compile specifically for Zen: - Zen/Zen2 are not included in the x86_64 or amd64 configurations. - Including them breaks AMD BLIS in other places, since some functions have Zen-specific definitions that fail on other platforms. Non-Zen CI runners will succeed when they are Haswell or later, since a Haswell kernel is compiled as a dependency.
This commit is contained in:
parent
386ebbccf0
commit
4fbe92df09
67
pkgs/development/libraries/science/math/amd-blis/default.nix
Normal file
67
pkgs/development/libraries/science/math/amd-blis/default.nix
Normal file
@ -0,0 +1,67 @@
|
|||||||
|
{ stdenv
|
||||||
|
, fetchFromGitHub
|
||||||
|
, perl
|
||||||
|
, python3
|
||||||
|
|
||||||
|
# Enable BLAS interface with 64-bit integer width.
|
||||||
|
, blas64 ? false
|
||||||
|
|
||||||
|
# Target architecture, use "zen" or "zen2", optimization for Zen and
|
||||||
|
# other families is pretty much mutually exclusive in the AMD fork of
|
||||||
|
# BLIS.
|
||||||
|
, withArchitecture ? "zen"
|
||||||
|
|
||||||
|
# Enable OpenMP-based threading.
|
||||||
|
, withOpenMP ? true
|
||||||
|
}:
|
||||||
|
|
||||||
|
let
|
||||||
|
threadingSuffix = if withOpenMP then "-mt" else "";
|
||||||
|
blasIntSize = if blas64 then "64" else "32";
|
||||||
|
in stdenv.mkDerivation rec {
|
||||||
|
pname = "amd-blis";
|
||||||
|
version = "2.2";
|
||||||
|
|
||||||
|
src = fetchFromGitHub {
|
||||||
|
owner = "amd";
|
||||||
|
repo = "blis";
|
||||||
|
rev = version;
|
||||||
|
sha256 = "1b2f5bwi0gkw2ih2rb7wfzn3m9hgg7k270kg43rmzpr2acpy86xa";
|
||||||
|
};
|
||||||
|
|
||||||
|
inherit blas64;
|
||||||
|
|
||||||
|
nativeBuildInputs = [
|
||||||
|
perl
|
||||||
|
python3
|
||||||
|
];
|
||||||
|
|
||||||
|
doCheck = true;
|
||||||
|
|
||||||
|
enableParallelBuilding = true;
|
||||||
|
|
||||||
|
configureFlags = [
|
||||||
|
"--enable-cblas"
|
||||||
|
"--blas-int-size=${blasIntSize}"
|
||||||
|
] ++ stdenv.lib.optionals withOpenMP [ "--enable-threading=openmp" ]
|
||||||
|
++ [ withArchitecture ];
|
||||||
|
|
||||||
|
postPatch = ''
|
||||||
|
patchShebangs configure build/flatten-headers.py
|
||||||
|
'';
|
||||||
|
|
||||||
|
postInstall = ''
|
||||||
|
ln -s $out/lib/libblis${threadingSuffix}.so.2 $out/lib/libblas.so.3
|
||||||
|
ln -s $out/lib/libblis${threadingSuffix}.so.2 $out/lib/libcblas.so.3
|
||||||
|
ln -s $out/lib/libblas.so.3 $out/lib/libblas.so
|
||||||
|
ln -s $out/lib/libcblas.so.3 $out/lib/libcblas.so
|
||||||
|
'';
|
||||||
|
|
||||||
|
meta = with stdenv.lib; {
|
||||||
|
description = "BLAS-compatible library optimized for AMD CPUs";
|
||||||
|
homepage = "https://developer.amd.com/amd-aocl/blas-library/";
|
||||||
|
license = licenses.bsd3;
|
||||||
|
maintainers = [ maintainers.danieldk ];
|
||||||
|
platforms = [ "x86_64-linux" ];
|
||||||
|
};
|
||||||
|
}
|
@ -25033,6 +25033,8 @@ in
|
|||||||
|
|
||||||
almonds = callPackage ../applications/science/math/almonds { };
|
almonds = callPackage ../applications/science/math/almonds { };
|
||||||
|
|
||||||
|
amd-blis = callPackage ../development/libraries/science/math/amd-blis { };
|
||||||
|
|
||||||
arpack = callPackage ../development/libraries/science/math/arpack { };
|
arpack = callPackage ../development/libraries/science/math/arpack { };
|
||||||
|
|
||||||
blas = callPackage ../build-support/alternatives/blas { };
|
blas = callPackage ../build-support/alternatives/blas { };
|
||||||
|
Loading…
Reference in New Issue
Block a user