python3Packages.cvxpy: add openmp multithreading

wip/yesman
Drew Risinger 3 years ago
parent 23e399366c
commit e44319a3e6
  1. 7
      pkgs/development/python-modules/cvxpy/default.nix

@ -9,6 +9,7 @@
, osqp
, scipy
, scs
, useOpenmp ? true
# Check inputs
, pytestCheckHook
}:
@ -34,6 +35,12 @@ buildPythonPackage rec {
scs
];
# Required flags from https://github.com/cvxgrp/cvxpy/releases/tag/v1.1.11
preBuild = lib.optional useOpenmp ''
export CFLAGS="-fopenmp"
export LDFLAGS="-lgomp"
'';
checkInputs = [ pytestCheckHook ];
pytestFlagsArray = [ "./cvxpy" ];
# Disable the slowest benchmarking tests, cuts test time in half

Loading…
Cancel
Save