zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update cuDNN packages to check Power installation (#19620)

Browse Source 
* Updated cuDNN package to check to make sure that target directory
exists before linking it.

* Fixed flake8

* Fixed Flake8
This commit is contained in:
Brian Van Essen 2020-11-02 18:08:44 -08:00 committed by GitHub
parent b4ea74c11b
commit d38fdb3854
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
var/spack/repos/builtin/packages/cudnn/package.py
View File

@ -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)