blob: 1c774b52910471eb1b88023716336843c5b126ad (
plain)
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
|
{
buildRedist,
cuda_nvrtc,
lib,
}:
buildRedist (finalAttrs: {
redistName = "cuda";
pname = "libcublas";
# libcublasLt dlopens NVRTC to compile kernels at runtime; absent before 12.8.
appendRunpaths = lib.optionals (lib.versionAtLeast finalAttrs.version "12.8") [
"${lib.getLib cuda_nvrtc}/lib" # libnvrtc.so.%s
];
outputs = [
"out"
"dev"
"include"
"lib"
"static"
"stubs"
];
meta = {
description = "CUDA Basic Linear Algebra Subroutine library";
longDescription = ''
The cuBLAS library is an implementation of BLAS (Basic Linear Algebra Subprograms) on top of the NVIDIA CUDA runtime.
'';
homepage = "https://developer.nvidia.com/cublas";
};
})
|
