diff options
Diffstat (limited to 'pkgs/test/cuda/cuda-library-samples/extension.nix')
-rw-r--r-- | pkgs/test/cuda/cuda-library-samples/extension.nix | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/pkgs/test/cuda/cuda-library-samples/extension.nix b/pkgs/test/cuda/cuda-library-samples/extension.nix index 4c721a9e9e1b..62de715fd0b4 100644 --- a/pkgs/test/cuda/cuda-library-samples/extension.nix +++ b/pkgs/test/cuda/cuda-library-samples/extension.nix @@ -1,3 +1,14 @@ -final: prev: { - cuda-library-samples = final.callPackage ./generic.nix { }; -} +{hostPlatform, lib}: +let + # Samples are built around the CUDA Toolkit, which is not available for + # aarch64. Check for both CUDA version and platform. + platformIsSupported = hostPlatform.isx86_64; + + # Build our extension + extension = + final: _: + lib.attrsets.optionalAttrs platformIsSupported { + cuda-library-samples = final.callPackage ./generic.nix {}; + }; +in +extension |