diff options
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index 76aa4db185..cea7e8006f 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -360,6 +360,11 @@ class PyTorch(PythonPackage, CudaPackage): 'intel-mkl', 'intel-parallel-studio', 'intel-oneapi-mkl']: env.set('BLAS', 'MKL') env.set('WITH_BLAS', 'mkl') + # help find MKL + if self.spec['mkl'].name == 'intel-oneapi-mkl': + env.set('INTEL_MKL_DIR', self.spec['mkl'].prefix.mkl.latest) + else: + env.set('INTEL_MKL_DIR', self.spec['mkl'].prefix.mkl) elif self.spec['blas'].name == 'openblas': env.set('BLAS', 'OpenBLAS') env.set('WITH_BLAS', 'open') |