···

       
35
       
35
       
 
       
, giflib

     

       
36
       
36
       
 
       
, grpc

     

       
37
       
37
       
 
       
, libjpeg_turbo

     

       
38
       
38
       
+
       
, protobuf

     

       
38
       
39
       
 
       
, python

     

       
39
       
40
       
 
       
, snappy

     

       
40
       
41
       
 
       
, zlib

     
···

       
49
       
50
       
 
       
}:

     

       
50
       
51
       
 
       


     

       
51
       
52
       
 
       
let

     

       
52
       
52
       
-
       


     

       
53
       
53
       
 
       
  inherit (cudaPackages) cudatoolkit cudnn nccl;

     

       
54
       
54
       
 
       


     

       
55
       
55
       
 
       
  pname = "jaxlib";

     
···

       
120
       
120
       
 
       
      numpy

     

       
121
       
121
       
 
       
      openssl

     

       
122
       
122
       
 
       
      pkgs.flatbuffers

     

       
123
       
123
       
-
       
      pkgs.protobuf

     

       
123
       
123
       
+
       
      protobuf

     

       
124
       
124
       
 
       
      pybind11

     

       
125
       
125
       
 
       
      scipy

     

       
126
       
126
       
 
       
      six

     
···

       
158
       
158
       
 
       
      build --action_env=PYENV_ROOT

     

       
159
       
159
       
 
       
      build --python_path="${python}/bin/python"

     

       
160
       
160
       
 
       
      build --distinct_host_configuration=false

     

       
161
       
161
       
-
       
      build --define PROTOBUF_INCLUDE_PATH="${pkgs.protobuf}/include"

     

       
161
       
161
       
+
       
      build --define PROTOBUF_INCLUDE_PATH="${protobuf}/include"

     

       
162
       
162
       
 
       
    '' + lib.optionalString cudaSupport ''

     

       
163
       
163
       
 
       
      build --action_env CUDA_TOOLKIT_PATH="${cudatoolkit_joined}"

     

       
164
       
164
       
 
       
      build --action_env CUDNN_INSTALL_PATH="${cudnn}"

     
···

       
268
       
268
       
 
       
        sed -i 's@-lprotobuf@-l:libprotobuf.a@' ../output/external/org_tensorflow/third_party/systemlibs/protobuf.BUILD

     

       
269
       
269
       
 
       
        sed -i 's@-lprotoc@-l:libprotoc.a@' ../output/external/org_tensorflow/third_party/systemlibs/protobuf.BUILD

     

       
270
       
270
       
 
       
      '' else if stdenv.cc.isClang then ''

     

       
271
       
271
       
-
       
        sed -i 's@-lprotobuf@${pkgs.protobuf}/lib/libprotobuf.a@' ../output/external/org_tensorflow/third_party/systemlibs/protobuf.BUILD

     

       
272
       
272
       
-
       
        sed -i 's@-lprotoc@${pkgs.protobuf}/lib/libprotoc.a@' ../output/external/org_tensorflow/third_party/systemlibs/protobuf.BUILD

     

       
271
       
271
       
+
       
        sed -i 's@-lprotobuf@${protobuf}/lib/libprotobuf.a@' ../output/external/org_tensorflow/third_party/systemlibs/protobuf.BUILD

     

       
272
       
272
       
+
       
        sed -i 's@-lprotoc@${protobuf}/lib/libprotoc.a@' ../output/external/org_tensorflow/third_party/systemlibs/protobuf.BUILD

     

       
273
       
273
       
 
       
      '' else throw "Unsupported stdenv.cc: ${stdenv.cc}");

     

       
274
       
274
       
 
       


     

       
275
       
275
       
 
       
      installPhase = ''

     

···

       
4699
       
4699
       
 
       
    # pin to `cudaPackages_11_6` instead.

     

       
4700
       
4700
       
 
       
    cudaPackages = pkgs.cudaPackages_11_6;

     

       
4701
       
4701
       
 
       
    IOKit = pkgs.darwin.apple_sdk_11_0.IOKit;

     

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

     

       
4702
       
4703
       
 
       
  };

     

       
4703
       
4704
       
 
       


     

       
4704
       
4705
       
 
       
  jaxlib = self.jaxlib-build;