25 lines
706 B
Nix
25 lines
706 B
Nix
|
{
|
||
|
|
cudaAtLeast,
|
||
|
|
lib,
|
||
|
|
libcublas,
|
||
|
|
libcusparse ? null,
|
||
|
|
libnvjitlink ? null,
|
||
|
|
}:
|
||
|
|
prevAttrs: {
|
||
|
|
buildInputs =
|
||
|
|
prevAttrs.buildInputs or [ ]
|
||
|
|
# Always depends on this
|
||
|
|
++ [ libcublas ]
|
||
|
|
# Dependency from 12.0 and on
|
||
|
|
++ lib.lists.optionals (cudaAtLeast "12.0") [ libnvjitlink ]
|
||
|
|
# Dependency from 12.1 and on
|
||
|
|
++ lib.lists.optionals (cudaAtLeast "12.1") [ libcusparse ];
|
||
|
|
|
||
|
|
brokenConditions = prevAttrs.brokenConditions or { } // {
|
||
|
|
"libnvjitlink missing (CUDA >= 12.0)" =
|
||
|
|
!(cudaAtLeast "12.0" -> (libnvjitlink != null && libnvjitlink != null));
|
||
|
|
"libcusparse missing (CUDA >= 12.1)" =
|
||
|
|
!(cudaAtLeast "12.1" -> (libcusparse != null && libcusparse != null));
|
||
|
|
};
|
||
|
|
}
|