Commit 9d42b468 authored by Connor Baker's avatar Connor Baker
Browse files

cudaPackages.libcudla: tidy expression for pre-CUDA 12

parent 33d8fe30
Loading
Loading
Loading
Loading
+3 −8
Original line number Diff line number Diff line
{
  backendStdenv,
  buildRedist,
  cudaMajorMinorVersion,
  cudaOlder,
  lib,
}:
buildRedist {
@@ -19,7 +19,7 @@ buildRedist {
  autoPatchelfIgnoreMissingDeps = [
    "libnvcudla.so"
  ]
  ++ lib.optionals (cudaMajorMinorVersion == "11.8") [
  ++ lib.optionals (cudaOlder "12") [
    "libcuda.so.1"
    "libnvdla_runtime.so"
  ];
@@ -27,12 +27,7 @@ buildRedist {
  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 == [ ]);
      assertion = lib.subtractLists [ "7.2" "8.7" ] backendStdenv.cudaCapabilities == [ ];
    }
  ];
}