pkgs/top-level/release-cuda.nix

   1 /*
   2     Test CUDA packages.
   3
   4     This release file will not be tested on hydra.nixos.org
   5     because it requires unfree software.
   6
   7     Test for example like this:
   8
   9         $ hydra-eval-jobs pkgs/top-level/release-cuda.nix --option restrict-eval false -I foo=. --arg nixpkgs '{ outPath = ./.; revCount = 0; shortRev = "aabbcc"; }'
  10
  11 */
  12
  13 { # The platforms for which we build Nixpkgs.
  14   supportedSystems ? [
  15     "x86_64-linux"
  16   ]
  17 , # Attributes passed to nixpkgs.
  18   nixpkgsArgs ? { config = { allowUnfree = true; inHydra = true; }; }
  19 }:
  20
  21 let
  22   release-lib = import ./release-lib.nix {
  23     inherit supportedSystems nixpkgsArgs;
  24   };
  25
  26   inherit (release-lib) linux mapTestOn packagePlatforms pkgs;
  27
  28   inherit (release-lib.lib) genAttrs;
  29
  30   # Package sets to evaluate
  31   packageSets = [
  32     "cudaPackages_10_0"
  33     "cudaPackages_10_1"
  34     "cudaPackages_10_2"
  35     "cudaPackages_10"
  36     "cudaPackages_11_0"
  37     "cudaPackages_11_1"
  38     "cudaPackages_11_2"
  39     "cudaPackages_11_3"
  40     "cudaPackages_11_4"
  41     "cudaPackages_11_5"
  42     "cudaPackages_11_6"
  43     "cudaPackages_11"
  44     "cudaPackages"
  45   ];
  46
  47   evalPackageSet = pset: mapTestOn { ${pset} = packagePlatforms pkgs.${pset}; };
  48
  49   jobs = (mapTestOn ({
  50     # Packages to evaluate
  51     python3.pkgs.caffeWithCuda = linux;
  52     python3.pkgs.jaxlibWithCuda = linux;
  53     python3.pkgs.libgpuarray = linux;
  54     python3.pkgs.tensorflowWithCuda = linux;
  55     python3.pkgs.pyrealsense2WithCuda = linux;
  56     python3.pkgs.torchWithCuda = linux;
  57     python3.pkgs.jaxlib = linux;
  58   }) // (genAttrs packageSets evalPackageSet));
  59
  60 in jobs