summaryrefslogtreecommitdiffstats
path: root/pkgs/development/cuda-modules/cuda-library-samples/extension.nix
blob: 456ab8168a459a8bf77cf040ab8a04a785a85b56 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
{ 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 && hostPlatform.isLinux;

  # Build our extension
  extension =
    final: _:
    lib.attrsets.optionalAttrs platformIsSupported {
      cuda-library-samples = final.callPackage ./generic.nix { };
    };
in
extension