diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index c0e3d012549..63f9218dc79 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 @@ def build_amd(self): 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