-
Notifications
You must be signed in to change notification settings - Fork 1
/
default.nix
56 lines (49 loc) · 986 Bytes
/
default.nix
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
{
config,
lib,
autoAddDriverRunpath,
cudaPackages,
...
}:
let
inherit (cudaPackages)
backendStdenv
cuda_cudart
cuda_nvcc
cudaAtLeast
cudaOlder
cudatoolkit
setupCudaHook
;
inherit (lib) getDev getLib getOutput;
in
backendStdenv.mkDerivation {
pname = "tdr-inverse";
version = "0.0.1";
src = ./.;
strictDeps = true;
nativeBuildInputs = [
autoAddDriverRunpath
cuda_nvcc
];
buildInputs = [ cuda_cudart ];
# buildPhase = ''
# make
# '';
installPhase = ''
mkdir -p $out/bin
install ./bin/inverse $out/bin
install ./bin/cpu-inverse $out/bin
install ./bin/tdr-inverse $out/bin
'';
meta = with lib; {
name = "GPU-Based Matrix Inversion";
description = ''
Optimized GPU-Based Matrix Inversion
Though The Use of Thread-Data Remapping
'';
homepage = "https://github.com/scrufulufugus/tdr-inverse";
license = licenses.bsd3;
platforms = platforms.all;
};
}