tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 _cuda,
3 buildRedist,
4 libcublas,
5 libnvshmem,
6 nccl,
7}:
8buildRedist {
9 redistName = "cublasmp";
10 pname = "libcublasmp";
11
12 outputs = [
13 "out"
14 "dev"
15 "include"
16 "lib"
17 ];
18
19 # TODO: Looks like the minimum supported capability is 7.0 as of the latest:
20 # https://docs.nvidia.com/cuda/cublasmp/getting_started/index.html
21 buildInputs = [
22 libcublas
23 libnvshmem
24 nccl
25 ];
26
27 autoPatchelfIgnoreMissingDeps = [
28 "libcuda.so.1"
29 ];
30
31 meta = {
32 description = "High-performance, multi-process, GPU-accelerated library for distributed basic dense linear algebra";
33 longDescription = ''
34 NVIDIA cuBLASMp is a high-performance, multi-process, GPU-accelerated library for distributed basic dense linear
35 algebra.
36
37 cuBLASMp is compatible with 2D block-cyclic data layout and provides PBLAS-like C APIs.
38 '';
39 homepage = "https://docs.nvidia.com/cuda/cublasmp";
40 changelog = "https://docs.nvidia.com/cuda/cublasmp/release_notes";
41 license = _cuda.lib.licenses.math_sdk_sla;
42 };
43}