13 lines
459 B
Nix
13 lines
459 B
Nix
|
{ flags, lib }:
|
||
|
|
prevAttrs: {
|
||
|
|
autoPatchelfIgnoreMissingDeps = prevAttrs.autoPatchelfIgnoreMissingDeps or [ ] ++ [
|
||
|
|
"libnvrm_gpu.so"
|
||
|
|
"libnvrm_mem.so"
|
||
|
|
"libnvdla_runtime.so"
|
||
|
|
];
|
||
|
|
# `cuda_compat` only works on aarch64-linux, and only when building for Jetson devices.
|
||
|
|
badPlatformsConditions = prevAttrs.badPlatformsConditions or { } // {
|
||
|
|
"Trying to use cuda_compat on aarch64-linux targeting non-Jetson devices" = !flags.isJetsonBuild;
|
||
|
|
};
|
||
|
|
}
|