pkgs/development/python-modules/dlib/default.nix

   1 { stdenv, buildPythonPackage, dlib, python, pytest, more-itertools
   2 , sse4Support ? stdenv.hostPlatform.sse4_1Support
   3 , avxSupport ? stdenv.hostPlatform.avxSupport
   4 }:
   5
   6 buildPythonPackage {
   7   inherit (dlib) pname version src nativeBuildInputs buildInputs meta;
   8
   9   format = "setuptools";
  10
  11   patches = [
  12     ./build-cores.patch
  13   ];
  14
  15   nativeCheckInputs = [ pytest more-itertools ];
  16
  17   postPatch = ''
  18     substituteInPlace setup.py \
  19       --replace "more-itertools<6.0.0" "more-itertools" \
  20       --replace "pytest==3.8" "pytest"
  21   '';
  22
  23   # although AVX can be enabled, we never test with it. Some Hydra machines
  24   # fail because of this, however their build results are probably used on hardware
  25   # with AVX support.
  26   checkPhase = ''
  27     ${python.interpreter} nix_run_setup test --no USE_AVX_INSTRUCTIONS
  28   '';
  29
  30   setupPyBuildFlags = [
  31     "--set USE_SSE4_INSTRUCTIONS=${if sse4Support then "yes" else "no"}"
  32     "--set USE_AVX_INSTRUCTIONS=${if avxSupport then "yes" else "no"}"
  33   ];
  34
  35   dontUseCmakeConfigure = true;
  36 }