From 33a34f9619a1eff866a1ae3b673f92d4ce5936a5 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Mon, 28 Oct 2019 11:47:21 -0500 Subject: [PATCH] Fix PyTorch build with Intel MKL (#13464) --- var/spack/repos/builtin/packages/py-torch/package.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index 451c71d7432..8fec773e3d5 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -169,6 +169,11 @@ def enable_or_disable(variant, keyword='USE', var=None, newer=False): else: env.set('NO_' + var, 'ON') + # Build system has problems locating MKL libraries + # See https://github.com/pytorch/pytorch/issues/24334 + if 'mkl' in self.spec: + env.prepend_path('CMAKE_PREFIX_PATH', self.spec['mkl'].prefix.mkl) + env.set('MAX_JOBS', make_jobs) enable_or_disable('cuda')