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 with import ./release-lib.nix {inherit supportedSystems nixpkgsArgs; };
  22 with lib;
  23
  24 let
  25   # Package sets to evaluate
  26   packageSets = [
  27     "cudaPackages_10_0"
  28     "cudaPackages_10_1"
  29     "cudaPackages_10_2"
  30     "cudaPackages_10"
  31     "cudaPackages_11_0"
  32     "cudaPackages_11_1"
  33     "cudaPackages_11_2"
  34     "cudaPackages_11_3"
  35     "cudaPackages_11_4"
  36     "cudaPackages_11_5"
  37     "cudaPackages_11_6"
  38     "cudaPackages_11"
  39     "cudaPackages"
  40   ];
  41
  42   evalPackageSet = pset: mapTestOn { ${pset} = packagePlatforms pkgs.${pset}; };
  43
  44   jobs = (mapTestOn ({
  45     # Packages to evaluate
  46     python3.pkgs.caffeWithCuda = linux;
  47     python3.pkgs.jaxlibWithCuda = linux;
  48     python3.pkgs.libgpuarray = linux;
  49     python3.pkgs.tensorflowWithCuda = linux;
  50     python3.pkgs.pyrealsense2WithCuda = linux;
  51     python3.pkgs.torchWithCuda = linux;
  52     python3.pkgs.jaxlib = linux;
  53   }) // (genAttrs packageSets evalPackageSet));
  54
  55 in jobs