pyrox.dev
1{
2 cudaAtLeast,
3 lib,
4 libcublas,
5 libcusparse ? null,
6 libnvjitlink ? null,
7}:
8prevAttrs: {
9 buildInputs = prevAttrs.buildInputs or [ ] ++ [
10 libcublas
11 libnvjitlink
12 libcusparse
13 ];
14
15 brokenConditions = prevAttrs.brokenConditions or { } // {
16 "libnvjitlink missing (CUDA >= 12.0)" = libnvjitlink == null;
17 "libcusparse missing (CUDA >= 12.1)" = libcusparse == null;
18 };
19}