diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-06-20 18:14:20 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-20 18:17:51 -0700 |
commit | f671c5caad24b3a107651dd840c043cdea03f0dd (patch) | |
tree | b9c5817366dcd91244d09d176aca426dfee0c237 /third_party/gpus/cuda | |
parent | 9403e4f8243606eead6d996100edcb4f872156f2 (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.tpl | 82 |
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"], ) |