public/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

python3Packages.cvxpy: add openmp multithreading

Browse Source
This commit is contained in:
Drew Risinger 2021-04-12 11:41:37 -04:00
parent 23e399366c
commit e44319a3e6
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pkgs/development/python-modules/cvxpy/default.nix
View File

@ -9,6 +9,7 @@
, osqp , osqp
, scipy , scipy
, scs , scs
, useOpenmp ? true
# Check inputs # Check inputs
, pytestCheckHook , pytestCheckHook
}: }:
@ -34,6 +35,12 @@ buildPythonPackage rec {
scs 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 ]; checkInputs = [ pytestCheckHook ];
pytestFlagsArray = [ "./cvxpy" ]; pytestFlagsArray = [ "./cvxpy" ];
# Disable the slowest benchmarking tests, cuts test time in half # Disable the slowest benchmarking tests, cuts test time in half