about summary refs log tree commit diff
path: root/nixpkgs/pkgs/development/cuda-modules/cudnn/shims.nix
diff options
context:
space:
mode:
Diffstat (limited to 'nixpkgs/pkgs/development/cuda-modules/cudnn/shims.nix')
-rw-r--r--nixpkgs/pkgs/development/cuda-modules/cudnn/shims.nix18
1 files changed, 13 insertions, 5 deletions
diff --git a/nixpkgs/pkgs/development/cuda-modules/cudnn/shims.nix b/nixpkgs/pkgs/development/cuda-modules/cudnn/shims.nix
index e9eca8ef7c8b..a36ee26dab5d 100644
--- a/nixpkgs/pkgs/development/cuda-modules/cudnn/shims.nix
+++ b/nixpkgs/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)";