diff options
| author | Connor Baker <connor.baker@tweag.io> | 2023-12-04 14:03:11 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-04 14:03:11 -0500 |
| commit | 5f5e76c08153f251bd7cd5dba20fe8254925597d (patch) | |
| tree | 3010f5748a05ec79c3e99134238dcb65ea88a5b9 /pkgs | |
| parent | f0124cff665c80ff983b62f1d3454e9be604fd74 (diff) | |
| parent | 2252b26260f64ab7d21bddd3b6d1ceb2564c30c2 (diff) | |
Merge pull request #272086 from ConnorBaker/fix/jaxlib-bin-remove-cuda-asserts
python3Packages.jaxlib-bin: move asserts to broken to unbreak eval
Diffstat (limited to 'pkgs')
| -rw-r--r-- | pkgs/development/python-modules/jaxlib/bin.nix | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/pkgs/development/python-modules/jaxlib/bin.nix b/pkgs/development/python-modules/jaxlib/bin.nix index 4030bd5e4041d..d80cbc2a60183 100644 --- a/pkgs/development/python-modules/jaxlib/bin.nix +++ b/pkgs/development/python-modules/jaxlib/bin.nix @@ -34,11 +34,7 @@ let inherit (cudaPackages) cudatoolkit cudnn; -in - -assert cudaSupport -> lib.versionAtLeast cudatoolkit.version "11.1" && lib.versionAtLeast cudnn.version "8.2" && stdenv.isLinux; -let version = "0.4.20"; inherit (python) pythonVersion; @@ -213,5 +209,9 @@ buildPythonPackage { license = licenses.asl20; maintainers = with maintainers; [ samuela ]; platforms = [ "aarch64-darwin" "x86_64-linux" "x86_64-darwin" ]; + broken = + !(cudaSupport -> (cudaPackages ? cudatoolkit) && lib.versionAtLeast cudatoolkit.version "11.1") + || !(cudaSupport -> (cudaPackages ? cudnn) && lib.versionAtLeast cudnn.version "8.2") + || !(cudaSupport -> stdenv.isLinux); }; } |