blob: 3276354240f4a910273032020db0456974c4030d (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
{ lib, stdenv, fetchFromGitHub, gfortran
, python3, util-linux, which
, enableStatic ? stdenv.hostPlatform.isStatic
}:
stdenv.mkDerivation rec {
pname = "libxsmm";
version = "1.17";
src = fetchFromGitHub {
owner = "libxsmm";
repo = "libxsmm";
rev = version;
sha256 = "sha256-s/NEFU4IwQPLyPLwMmrrpMDd73q22Sk2BNid/kedawY=";
};
# Fixes /build references in the rpath
patches = [ ./rpath.patch ];
outputs = [ "out" "dev" "doc" ];
nativeBuildInputs = [
gfortran
python3
util-linux
which
];
enableParallelBuilding = true;
dontConfigure = true;
makeFlags = let
static = if enableStatic then "1" else "0";
in [
"OMP=1"
"PREFIX=$(out)"
"STATIC=${static}"
];
postInstall = ''
mkdir -p $dev/lib/pkgconfig
mv $out/lib/*.pc $dev/lib/pkgconfig
moveToOutput "share/libxsmm" ''${!outputDoc}
'';
prePatch = ''
patchShebangs .
'';
meta = with lib; {
broken = (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64);
description = "Library targeting Intel Architecture for specialized dense and sparse matrix operations, and deep learning primitives";
mainProgram = "libxsmm_gemm_generator";
license = licenses.bsd3;
homepage = "https://github.com/hfp/libxsmm";
platforms = platforms.linux;
maintainers = with lib.maintainers; [ chessai ];
};
}
|
