diff --git a/eb_hooks.py b/eb_hooks.py index 60c5b74f..9c7b835c 100644 --- a/eb_hooks.py +++ b/eb_hooks.py @@ -717,13 +717,18 @@ def pre_fetch_hook(self, *args, **kwargs): def pre_fetch_hook_check_installation_path(self, *args, **kwargs): # When we know the CUDA status, we will need to verify the installation path # if we are doing an EESSI or host_injections installation - accelerator_deps = ['CUDA'] + accelerator_deps = ['CUDA', 'ROCm-LLVM'] + accelerator_toolchains = ['rocm-compilers', 'rompi', 'rfbf', 'rfoss'] strict_eessi_installation = ( bool(re.search(EESSI_INSTALLATION_REGEX, self.installdir)) or self.installdir.startswith(HOST_INJECTIONS_LOCATION)) if strict_eessi_installation and not os.getenv("EESSI_OVERRIDE_STRICT_INSTALLPATH_CHECK"): dependency_names = self.cfg.dependency_names() - if self.cfg.name in accelerator_deps or any(dep in dependency_names for dep in accelerator_deps): + if ( + self.cfg.name in accelerator_deps + or any(dep in dependency_names for dep in accelerator_deps) + or self.toolchain.name in accelerator_toolchains + ): # Make sure the path is an accelerator location if "/accel/" not in self.installdir: raise EasyBuildError(