diff options
-rw-r--r-- | var/spack/repos/builtin/packages/hip/package.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/var/spack/repos/builtin/packages/hip/package.py b/var/spack/repos/builtin/packages/hip/package.py index 8b9b39f9a7..41c2d2d29f 100644 --- a/var/spack/repos/builtin/packages/hip/package.py +++ b/var/spack/repos/builtin/packages/hip/package.py @@ -288,10 +288,12 @@ class Hip(CMakePackage): if self.spec.external: # For external packages we only assume the `hip` prefix is known, # because spack does not set prefixes of dependencies of externals. - # We assume self.spec.prefix is /opt/rocm-x.y.z/hip and rocm has a - # default installation with everything installed under - # /opt/rocm-x.y.z - rocm_prefix = Prefix(os.path.dirname(self.spec.prefix)) + # We assume self.spec.prefix is /opt/rocm-x.y.z for rocm-5.2.0 and newer + # and /opt/rocm-x.y.z/hip for older versions + if self.spec.satisfies("@5.2.0:"): + rocm_prefix = Prefix(self.spec.prefix) + else: + rocm_prefix = Prefix(os.path.dirname(self.spec.prefix)) if not os.path.isdir(rocm_prefix): msg = "Could not determine prefix for other rocm components\n" |