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')