mirror of
https://github.com/ilyakooo0/nixpkgs.git
synced 2024-09-24 14:18:13 +03:00
pythonPackages.pytorch: fix Intel MKL BLAS compile failure
Signed-off-by: Austin Seipp <aseipp@pobox.com>
This commit is contained in:
parent
60176e005d
commit
af53114a09
@ -72,6 +72,11 @@ in buildPythonPackage rec {
|
||||
PYTORCH_BUILD_VERSION = version;
|
||||
PYTORCH_BUILD_NUMBER = 0;
|
||||
|
||||
# Suppress a weird warning in mkl-dnn, part of ideep in pytorch
|
||||
# (upstream seems to have fixed this in the wrong place?)
|
||||
# https://github.com/intel/mkl-dnn/commit/8134d346cdb7fe1695a2aa55771071d455fae0bc
|
||||
NIX_CFLAGS_COMPILE = lib.optionals (numpy.blasImplementation == "mkl") [ "-Wno-error=array-bounds" ];
|
||||
|
||||
buildInputs = [
|
||||
cmake
|
||||
numpy.blas
|
||||
|
Loading…
Reference in New Issue
Block a user