diff options
| author |  Gunhan Gulsoy <gunan@google.com> | 2018-09-26 01:38:55 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-26 01:43:15 -0700 |
| commit | ffa90fc521c6051addd50236872a4afaa45e0a49 (patch) | |
| tree | 4485cad4b3d97253fe0ef60cfae381041f9fba4b /third_party | |
| parent | c63d21b0bfc534b6377b332e9d2ba2abbdb7e0eb (diff) | |
Fixes for building with CUDA on ppc64le.
PiperOrigin-RevId: 214569615
Diffstat (limited to 'third_party')
| -rw-r--r-- | third_party/gpus/cuda_configure.bzl | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/third_party/gpus/cuda_configure.bzl b/third_party/gpus/cuda_configure.bzl index 5648b1525a..f5fdd3a75e 100644 --- a/third_party/gpus/cuda_configure.bzl +++ b/third_party/gpus/cuda_configure.bzl @@ -48,6 +48,7 @@ _DEFAULT_CUDA_COMPUTE_CAPABILITIES = ["3.5", "5.2"] CUDA_LIB_PATHS = [ "lib64/", "lib64/stubs/", + "lib/powerpc64le-linux-gnu/", "lib/x86_64-linux-gnu/", "lib/x64/", "lib/", @@ -70,6 +71,7 @@ CUPTI_HEADER_PATHS = [ # the other CUDA libraries but rather in a special extras/CUPTI directory. CUPTI_LIB_PATHS = [ "extras/CUPTI/lib64/", + "lib/powerpc64le-linux-gnu/", "lib/x86_64-linux-gnu/", "lib64/", "extras/CUPTI/libx64/", |