about summary refs log tree commit diff
path: root/pkgs/development/cuda-modules/cudnn/shims.nix
diff options
context:
space:
mode:
authorConnor Baker <connor.baker@tweag.io>2023-12-14 22:19:02 +0000
committerConnor Baker <connor.baker@tweag.io>2024-01-10 01:15:01 +0000
commit9bebd9e72d6b552fcfd3d1e6716eca6563944f42 (patch)
treed0b45b861d0d47be9e43c26481f68ce53a3d6987 /pkgs/development/cuda-modules/cudnn/shims.nix
parent501a1af970ca54cb300474a00aacfbd01f8a5b24 (diff)
tree-wide: cudaPackages should not break default eval
cudaPackages: guard expressions against null values
Diffstat (limited to 'pkgs/development/cuda-modules/cudnn/shims.nix')
-rw-r--r--pkgs/development/cuda-modules/cudnn/shims.nix18
1 files changed, 13 insertions, 5 deletions
diff --git a/pkgs/development/cuda-modules/cudnn/shims.nix b/pkgs/development/cuda-modules/cudnn/shims.nix
index e9eca8ef7c8b9..a36ee26dab5dc 100644
--- a/pkgs/development/cuda-modules/cudnn/shims.nix
+++ b/pkgs/development/cuda-modules/cudnn/shims.nix
@@ -1,10 +1,18 @@
 # Shims to mimic the shape of ../modules/generic/manifests/{feature,redistrib}/release.nix
-{package, redistArch}:
 {
-  featureRelease.${redistArch}.outputs = {
-    lib = true;
-    static = true;
-    dev = true;
+  lib,
+  package,
+  # redistArch :: String
+  # String is "unsupported" if the given architecture is unsupported.
+  redistArch,
+}:
+{
+  featureRelease = lib.optionalAttrs (redistArch != "unsupported") {
+    ${redistArch}.outputs = {
+      lib = true;
+      static = true;
+      dev = true;
+    };
   };
   redistribRelease = {
     name = "NVIDIA CUDA Deep Neural Network library (cuDNN)";