From adcfdf7eafd5267403c1fbe9f17f3df2c43f0f9b Mon Sep 17 00:00:00 2001
From: Connor Baker <ConnorBaker01@gmail.com>
Date: Fri, 3 Oct 2025 10:12:42 -0700
Subject: [PATCH] cudaPackages.libcudla: only available on Xavier and Orin

Signed-off-by: Connor Baker <ConnorBaker01@gmail.com>
---
 pkgs/development/cuda-modules/packages/libcudla.nix | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/pkgs/development/cuda-modules/packages/libcudla.nix b/pkgs/development/cuda-modules/packages/libcudla.nix
index c8a1b4549dcb..809c0008884d 100644
--- a/pkgs/development/cuda-modules/packages/libcudla.nix
+++ b/pkgs/development/cuda-modules/packages/libcudla.nix
@@ -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 == [ ]);
+    }
+  ];
 }