From 33a34f9619a1eff866a1ae3b673f92d4ce5936a5 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Mon, 28 Oct 2019 11:47:21 -0500 Subject: Fix PyTorch build with Intel MKL (#13464) --- var/spack/repos/builtin/packages/py-torch/package.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index 451c71d743..8fec773e3d 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 @@ class PyTorch(PythonPackage): 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') -- cgit v1.2.3-70-g09d2