From 9055deea16cdfb8344f2663a61e4603901110e72 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Tue, 6 Jul 2021 01:53:05 -0500 Subject: [PATCH] py-torch: fix bug in libs/headers attributes (#24624) --- var/spack/repos/builtin/packages/py-torch/package.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index 065deb08d2e..75125cd08c1 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -223,12 +223,14 @@ class PyTorch(PythonPackage, CudaPackage): @property def libs(self): - root = join_path(site_packages_dir, 'torch', 'lib') + root = join_path(self.prefix, self.spec['python'].package.site_packages_dir, + 'torch', 'lib') return find_libraries('libtorch', root) @property def headers(self): - root = join_path(site_packages_dir, 'torch', 'include') + root = join_path(self.prefix, self.spec['python'].package.site_packages_dir, + 'torch', 'include') headers = find_all_headers(root) headers.directories = [root] return headers