cuda: add installation workaround for version 10.1.243 on ppc64le (#21200)

Fixes #21170
This commit is contained in:
Baptiste Jonglez 2021-01-22 23:27:14 +01:00 committed by GitHub
parent cdb9b10e27
commit 49f0e9640d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -134,6 +134,15 @@ def install(self, spec, prefix):
# https://gist.github.com/ax3l/9489132 # https://gist.github.com/ax3l/9489132
# for details. # for details.
# CUDA 10.1 on ppc64le fails to copy some files, the workaround is adapted from
# https://forums.developer.nvidia.com/t/cuda-10-1-243-10-1-update-2-ppc64le-run-file-installation-issue/82433
# See also #21170
if spec.satisfies('@10.1.243') and platform.machine() == 'ppc64le':
includedir = "targets/ppc64le-linux/include"
os.makedirs(os.path.join(prefix, includedir))
os.makedirs(os.path.join(prefix, "src"))
os.symlink(includedir, os.path.join(prefix, "include"))
# CUDA 10.1+ has different cmdline options for the installer # CUDA 10.1+ has different cmdline options for the installer
arguments = [ arguments = [
runfile, # the install script runfile, # the install script