python3Packages.cvxpy: add openmp multithreading

This commit is contained in:
Drew Risinger 2021-04-12 11:41:37 -04:00
parent 23e399366c
commit e44319a3e6

View file

@ -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