Fix PyTorch build with Intel MKL (#13464)

2019-10-28 11:47:21 -05:00
parent b4e027e918
commit 33a34f9619
1 changed files with 5 additions and 0 deletions
--- 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')