commit d794669e128db518d2ea24dc3ea1d82cae16dafe · pyrox.dev/nixpkgs

+23 -8
pkgs/development/python-modules/dlib/default.nix
···

       1
       1
        
       {

     

       2
       2
       -
         stdenv,

     

       3
       2
        
         buildPythonPackage,

     

       4
       3
        
         dlib,

     

       5
       5
       -
         python,

     

       6
       4
        
         pytestCheckHook,

     

       7
       5
        
         more-itertools,

     

       8
       8
       -
         sse4Support ? stdenv.hostPlatform.sse4_1Support,

     

       9
       9
       -
         avxSupport ? stdenv.hostPlatform.avxSupport,

     

       10
       6
        
       }:

     

       11
       7
        
       

     

       12
       8
        
       buildPythonPackage {

     

       13
       9
        
         inherit (dlib)

     

       10
       10
       +
           stdenv

     

       14
       11
        
           pname

     

       15
       12
        
           version

     

       16
       13
        
           src

     

       17
       14
        
           nativeBuildInputs

     

       18
       15
        
           buildInputs

     

       16
       16
       +
           cmakeFlags

     

       17
       17
       +
           passthru

     

       19
       18
        
           meta

     

       20
       19
        
           ;

     

       21
       20
        
       

     
···

       34
       33
        
             --replace "pytest==3.8" "pytest"

     

       35
       34
        
         '';

     

       36
       35
        
       

     

       37
       37
       -
         setupPyBuildFlags = [

     

       38
       38
       -
           "--set USE_SSE4_INSTRUCTIONS=${if sse4Support then "yes" else "no"}"

     

       39
       39
       -
           "--set USE_AVX_INSTRUCTIONS=${if avxSupport then "yes" else "no"}"

     

       40
       40
       -
         ];

     

       36
       36
       +
         # Pass CMake flags through to the build script

     

       37
       37
       +
         preConfigure = ''

     

       38
       38
       +
           for flag in $cmakeFlags; do

     

       39
       39
       +
             if [[ "$flag" == -D* ]]; then

     

       40
       40
       +
               setupPyBuildFlags+=" --set ''${flag#-D}"

     

       41
       41
       +
             fi

     

       42
       42
       +
           done

     

       43
       43
       +
         '';

     

       41
       44
        
       

     

       42
       45
        
         dontUseCmakeConfigure = true;

     

       46
       46
       +
       

     

       47
       47
       +
         doCheck =

     

       48
       48
       +
           !(

     

       49
       49
       +
             # The tests attempt to use CUDA on the build platform.

     

       50
       50
       +
             # https://github.com/NixOS/nixpkgs/issues/225912

     

       51
       51
       +
             dlib.cudaSupport

     

       52
       52
       +
       

     

       53
       53
       +
             # although AVX can be enabled, we never test with it. Some Hydra machines

     

       54
       54
       +
             # fail because of this, however their build results are probably used on hardware

     

       55
       55
       +
             # with AVX support.

     

       56
       56
       +
             || dlib.avxSupport

     

       57
       57
       +
           );

     

       43
       58
        
       }