Commit 3fa9f1f0 authored by Samuel Ainsworth's avatar Samuel Ainsworth
Browse files

python3Packages.jaxlib-bin: fix ofborg evaluation 

See https://discourse.nixos.org/t/ofborg-does-not-respect-meta-platforms/27019 for more info.
parent 11687d06

pkgs/development/python-modules/jaxlib/bin.nix

+7 −1
Original line number Diff line number Diff line
@@ -77,7 +77,13 @@ buildPythonPackage rec { 
  # python version.

  disabled = !(pythonVersion == "3.10");



  src = if !cudaSupport then cpuSrcs."${stdenv.hostPlatform.system}" else gpuSrc;

  # See https://discourse.nixos.org/t/ofborg-does-not-respect-meta-platforms/27019/6.

  src =

    if !cudaSupport then

      (

        cpuSrcs."${stdenv.hostPlatform.system}"

          or (throw "jaxlib-bin is not supported on ${stdenv.hostPlatform.system}")

      ) else gpuSrc;



  # Prebuilt wheels are dynamically linked against things that nix can't find.

  # Run `autoPatchelfHook` to automagically fix them.