From 209f4678f2639ead6b93a3a61687ab2b8166f1ea Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Sat, 10 Apr 2021 08:09:38 +0200 Subject: [PATCH] python3Packages.pytorch: disable building of tests - We do not run them. - The test binaries are installed into site-packages by default. --- pkgs/development/python-modules/pytorch/default.nix | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pkgs/development/python-modules/pytorch/default.nix b/pkgs/development/python-modules/pytorch/default.nix index 225bba02c61c..1c0cc1b24f3a 100644 --- a/pkgs/development/python-modules/pytorch/default.nix +++ b/pkgs/development/python-modules/pytorch/default.nix @@ -164,6 +164,9 @@ in buildPythonPackage rec { BUILD_NAMEDTENSOR = setBool true; BUILD_DOCS = setBool buildDocs; + # We only do an imports check, so do not build tests either. + BUILD_TEST = setBool false; + USE_MKL = setBool (blas.implementation == "mkl"); # Unlike MKL, oneDNN (née MKLDNN) is FOSS, so we enable support for