tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 backendStdenv,
3 buildRedist,
4 cudaOlder,
5 lib,
6}:
7buildRedist {
8 redistName = "cuda";
9 pname = "libcudla";
10
11 outputs = [
12 "out"
13 "dev"
14 "include"
15 "lib"
16 "stubs"
17 ];
18
19 autoPatchelfIgnoreMissingDeps = [
20 "libnvcudla.so"
21 ]
22 ++ lib.optionals (cudaOlder "12") [
23 "libcuda.so.1"
24 "libnvdla_runtime.so"
25 ];
26
27 platformAssertions = [
28 {
29 message = "Only Xavier (7.2) and Orin (8.7) Jetson devices are supported";
30 assertion = lib.subtractLists [ "7.2" "8.7" ] backendStdenv.cudaCapabilities == [ ];
31 }
32 ];
33}