From c6b69a8244c10eef55c4bf11cf9251bfd1409da2 Mon Sep 17 00:00:00 2001 From: Glenn Johnson Date: Sun, 20 Mar 2022 10:35:18 -0500 Subject: 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. --- 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 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') -- cgit v1.2.3-70-g09d2