Gaetan Lepage
41 lines
876 B
Nix
41 lines
876 B
Nix
{
|
|
_cuda,
|
|
buildRedist,
|
|
cuda_cudart,
|
|
libcublas,
|
|
libcusolver,
|
|
nccl,
|
|
}:
|
|
buildRedist {
|
|
redistName = "cusolvermp";
|
|
pname = "libcusolvermp";
|
|
|
|
outputs = [
|
|
"out"
|
|
"dev"
|
|
"include"
|
|
"lib"
|
|
];
|
|
|
|
buildInputs = [
|
|
cuda_cudart
|
|
libcublas
|
|
libcusolver
|
|
nccl
|
|
];
|
|
|
|
autoPatchelfIgnoreMissingDeps = [
|
|
# Needs to be dynamically loaded as it depends on the hardware
|
|
"libcuda.so.1"
|
|
];
|
|
|
|
meta = {
|
|
description = "High-performance, distributed-memory, GPU-accelerated library that provides tools for solving dense linear systems and eigenvalue problems";
|
|
longDescription = ''
|
|
The NVIDIA cuSOLVERMp library is a high-performance, distributed-memory, GPU-accelerated library that provides
|
|
tools for solving dense linear systems and eigenvalue problems.
|
|
'';
|
|
homepage = "https://developer.nvidia.com/cusolver";
|
|
};
|
|
}
|