Commit bbaad9c2 authored by Samuel Ainsworth's avatar Samuel Ainsworth
Browse files

python3Packages.jaxlib{-bin}: no longer need to pin the cudaPackages version

parent d9a6568d

pkgs/top-level/python-packages.nix

+0 −6
Original line number Diff line number Diff line
@@ -4780,9 +4780,6 @@ in { 


  jaxlib-bin = callPackage ../development/python-modules/jaxlib/bin.nix {

    cudaSupport = pkgs.config.cudaSupport or false;

    # At the time of writing (2022-04-18), `cudaPackages.nccl` is broken, so we

    # pin to `cudaPackages_11_6` instead.

    cudaPackages = pkgs.cudaPackages_11_6;

  };



  jaxlib-build = callPackage ../development/python-modules/jaxlib rec {
@@ -4792,9 +4789,6 @@ in { 
    };

    # Some platforms don't have `cudaSupport` defined, hence the need for 'or false'.

    cudaSupport = pkgs.config.cudaSupport or false;

    # At the time of writing (2022-04-18), `cudaPackages.nccl` is broken, so we

    # pin to `cudaPackages_11_6` instead.

    cudaPackages = pkgs.cudaPackages_11_6;

    IOKit = pkgs.darwin.apple_sdk_11_0.IOKit;

    protobuf = pkgs.protobuf3_20; # jaxlib-build 0.3.15 won't build with protobuf 3.21

  };