From c6b69a8244c10eef55c4bf11cf9251bfd1409da2 Mon Sep 17 00:00:00 2001 From: Glenn Johnson Date: Sun, 20 Mar 2022 10:35:18 -0500 Subject: [PATCH] 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(+) diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index 76aa4db185c..cea7e8006f7 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 @@ def enable_or_disable(variant, keyword='USE', var=None, newer=False): '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')