From d38fdb3854cbcada368efd48c3faf44dea960e2f Mon Sep 17 00:00:00 2001 From: Brian Van Essen Date: Mon, 2 Nov 2020 18:08:44 -0800 Subject: [PATCH] Update cuDNN packages to check Power installation (#19620) * Updated cuDNN package to check to make sure that target directory exists before linking it. * Fixed flake8 * Fixed Flake8 --- var/spack/repos/builtin/packages/cudnn/package.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/var/spack/repos/builtin/packages/cudnn/package.py b/var/spack/repos/builtin/packages/cudnn/package.py index b8e45ce4b6d..7b9e3105243 100644 --- a/var/spack/repos/builtin/packages/cudnn/package.py +++ b/var/spack/repos/builtin/packages/cudnn/package.py @@ -257,8 +257,12 @@ def install(self, spec, prefix): install_tree('.', prefix) if 'target=ppc64le: platform=linux' in spec: - symlink(os.path.join(prefix, 'targets', 'ppc64le-linux', 'lib'), - prefix.lib) - symlink( - os.path.join(prefix, 'targets', 'ppc64le-linux', 'include'), - prefix.include) + target_lib = os.path.join(prefix, 'targets', + 'ppc64le-linux', 'lib') + if os.path.isdir(target_lib) and not os.path.isdir(prefix.lib): + symlink(target_lib, prefix.lib) + target_include = os.path.join(prefix, 'targets', + 'ppc64le-linux', 'include') + if os.path.isdir(target_include) \ + and not os.path.isdir(prefix.include): + symlink(target_include, prefix.include)