diff options
Diffstat (limited to 'var')
-rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 7 |
1 files changed, 7 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 c0e3d01254..63f9218dc7 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -606,3 +606,10 @@ class PyTorch(PythonPackage, CudaPackage, ROCmPackage): def install_test(self): with working_dir("test"): python("run_test.py") + + @property + def cmake_prefix_paths(self): + cmake_prefix_paths = [ + join_path(self.prefix, self.spec["python"].package.platlib, "torch", "share", "cmake") + ] + return cmake_prefix_paths |