python3Packages.cvxpy: add openmp multithreading
This commit is contained in:
parent
23e399366c
commit
e44319a3e6
1 changed files with 7 additions and 0 deletions
|
@ -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…
Reference in a new issue