Connor Baker
34 lines
558 B
Nix
34 lines
558 B
Nix
{
|
|
backendStdenv,
|
|
buildRedist,
|
|
cudaOlder,
|
|
lib,
|
|
}:
|
|
buildRedist {
|
|
redistName = "cuda";
|
|
pname = "libcudla";
|
|
|
|
outputs = [
|
|
"out"
|
|
"dev"
|
|
"include"
|
|
"lib"
|
|
"stubs"
|
|
];
|
|
|
|
autoPatchelfIgnoreMissingDeps = [
|
|
"libnvcudla.so"
|
|
]
|
|
++ lib.optionals (cudaOlder "12") [
|
|
"libcuda.so.1"
|
|
"libnvdla_runtime.so"
|
|
];
|
|
|
|
platformAssertions = [
|
|
{
|
|
message = "Only Xavier (7.2) and Orin (8.7) Jetson devices are supported";
|
|
assertion = lib.subtractLists [ "7.2" "8.7" ] backendStdenv.cudaCapabilities == [ ];
|
|
}
|
|
];
|
|
}
|