aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party/gpus/cuda
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-06-20 18:14:20 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-06-20 18:17:51 -0700
commitf671c5caad24b3a107651dd840c043cdea03f0dd (patch)
treeb9c5817366dcd91244d09d176aca426dfee0c237 /third_party/gpus/cuda
parent9403e4f8243606eead6d996100edcb4f872156f2 (diff)
Automated g4 rollback of changelist 159570549
PiperOrigin-RevId: 159643278
Diffstat (limited to 'third_party/gpus/cuda')
-rw-r--r--third_party/gpus/cuda/BUILD.tpl82
1 files changed, 27 insertions, 55 deletions
diff --git a/third_party/gpus/cuda/BUILD.tpl b/third_party/gpus/cuda/BUILD.tpl
index 2cca438d6d..f7610dd7a9 100644
--- a/third_party/gpus/cuda/BUILD.tpl
+++ b/third_party/gpus/cuda/BUILD.tpl
@@ -40,23 +40,20 @@ config_setting(
cc_library(
name = "cuda_headers",
hdrs = [
- "cuda/cuda_config.h",
+ "cuda_config.h",
%{cuda_headers}
],
includes = [
".",
- "cuda/include",
+ "include",
],
visibility = ["//visibility:public"],
)
cc_library(
name = "cudart_static",
- srcs = ["cuda/lib/%{cudart_static_lib}"],
- includes = [
- ".",
- "cuda/include",
- ],
+ srcs = ["lib/%{cudart_static_lib}"],
+ includes = ["include"],
linkopts = select({
":freebsd": [],
"//conditions:default": ["-ldl"],
@@ -69,46 +66,34 @@ cc_library(
cc_library(
name = "cuda_driver",
- srcs = ["cuda/lib/%{cuda_driver_lib}"],
- includes = [
- ".",
- "cuda/include",
- ],
+ srcs = ["lib/%{cuda_driver_lib}"],
+ includes = ["include"],
visibility = ["//visibility:public"],
)
cc_library(
name = "cudart",
- srcs = ["cuda/lib/%{cudart_lib}"],
- data = ["cuda/lib/%{cudart_lib}"],
- includes = [
- ".",
- "cuda/include",
- ],
+ srcs = ["lib/%{cudart_lib}"],
+ data = ["lib/%{cudart_lib}"],
+ includes = ["include"],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "cublas",
- srcs = ["cuda/lib/%{cublas_lib}"],
- data = ["cuda/lib/%{cublas_lib}"],
- includes = [
- ".",
- "cuda/include",
- ],
+ srcs = ["lib/%{cublas_lib}"],
+ data = ["lib/%{cublas_lib}"],
+ includes = ["include"],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "cusolver",
- srcs = ["cuda/lib/%{cusolver_lib}"],
- data = ["cuda/lib/%{cusolver_lib}"],
- includes = [
- ".",
- "cuda/include",
- ],
+ srcs = ["lib/%{cusolver_lib}"],
+ data = ["lib/%{cusolver_lib}"],
+ includes = ["include"],
linkstatic = 1,
linkopts = ["-lgomp"],
visibility = ["//visibility:public"],
@@ -116,36 +101,27 @@ cc_library(
cc_library(
name = "cudnn",
- srcs = ["cuda/lib/%{cudnn_lib}"],
- data = ["cuda/lib/%{cudnn_lib}"],
- includes = [
- ".",
- "cuda/include",
- ],
+ srcs = ["lib/%{cudnn_lib}"],
+ data = ["lib/%{cudnn_lib}"],
+ includes = ["include"],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "cufft",
- srcs = ["cuda/lib/%{cufft_lib}"],
- data = ["cuda/lib/%{cufft_lib}"],
- includes = [
- ".",
- "cuda/include",
- ],
+ srcs = ["lib/%{cufft_lib}"],
+ data = ["lib/%{cufft_lib}"],
+ includes = ["include"],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "curand",
- srcs = ["cuda/lib/%{curand_lib}"],
- data = ["cuda/lib/%{curand_lib}"],
- includes = [
- ".",
- "cuda/include",
- ],
+ srcs = ["lib/%{curand_lib}"],
+ data = ["lib/%{curand_lib}"],
+ includes = ["include"],
linkstatic = 1,
visibility = ["//visibility:public"],
)
@@ -166,23 +142,19 @@ cc_library(
cc_library(
name = "cupti_headers",
hdrs = [
- "cuda/cuda_config.h",
+ "cuda_config.h",
":cuda-extras",
],
includes = [
".",
- "cuda/extras/CUPTI/include/",
+ "extras/CUPTI/include/",
],
visibility = ["//visibility:public"],
)
cc_library(
name = "cupti_dsos",
- data = ["cuda/lib/%{cupti_lib}"],
- includes = [
- ".",
- "cuda/include",
- ],
+ data = ["lib/%{cupti_lib}"],
visibility = ["//visibility:public"],
)