1{
2 cudaAtLeast,
3 lib,
4 libnvjitlink ? null,
5}:
6prevAttrs: {
7 buildInputs =
8 prevAttrs.buildInputs or [ ]
9 # Dependency from 12.0 and on
10 ++ lib.lists.optionals (cudaAtLeast "12.0") [ libnvjitlink ];
11
12 brokenConditions = prevAttrs.brokenConditions or { } // {
13 "libnvjitlink missing (CUDA >= 12.0)" =
14 !(cudaAtLeast "12.0" -> (libnvjitlink != null && libnvjitlink != null));
15 };
16}