diff options
author | Glenn Johnson <glenn-johnson@uiowa.edu> | 2022-03-20 10:35:18 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-03-20 10:35:18 -0500 |
commit | c6b69a8244c10eef55c4bf11cf9251bfd1409da2 (patch) | |
tree | c88de051c20e63f60d1757b32dfa60d279faad17 | |
parent | d349677dcbb5fb1cbec405a4c6c4d6fc1a70ebbd (diff) | |
download | spack-c6b69a8244c10eef55c4bf11cf9251bfd1409da2.tar.gz spack-c6b69a8244c10eef55c4bf11cf9251bfd1409da2.tar.bz2 spack-c6b69a8244c10eef55c4bf11cf9251bfd1409da2.tar.xz spack-c6b69a8244c10eef55c4bf11cf9251bfd1409da2.zip |
py-torch: set INTEL_MKL_DIR (#29613)
Add the INTEL_MKL_DIR variable so py-torch can find mkl if provided by
intel-oneapi-mkl.
-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') |