cudaPackages.libcudla: only available on Xavier and Orin

Signed-off-by: Connor Baker <ConnorBaker01@gmail.com>
2025-10-22 13:14:51 -07:00
parent d21cad3bb1
commit adcfdf7eaf
1 changed files with 12 additions and 0 deletions
@@ -18,4 +18,16 @@ buildRedist {
  autoPatchelfIgnoreMissingDeps = lib.optionals backendStdenv.hasJetsonCudaCapability [
    "libnvcudla.so"
  ];
+
+  passthru.platformAssertions = [
+    {
+      message = "Only Xavier (7.2) and Orin (8.7) Jetson devices are supported";
+      assertion =
+        let
+          inherit (backendStdenv) hasJetsonCudaCapability requestedJetsonCudaCapabilities;
+        in
+        hasJetsonCudaCapability
+        -> (lib.subtractLists [ "7.2" "8.7" ] requestedJetsonCudaCapabilities == [ ]);
+    }
+  ];
 }