diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-07-04 08:04:25 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-04 08:10:41 -0700 |
commit | 9ec810573bd3666e36f960923a8aff7cb72df1dc (patch) | |
tree | b385c0665a81621dbe7eae69e4e9ec45e47f6418 /third_party/gpus/cuda | |
parent | a7032f21d72dd051f09b94733ed890dcd7ceaac8 (diff) |
Symlink files for cuda configuration one level below the BUILD file (i.e., one level under local_config_cuda/cuda).
END_PUBLIC
BEGIN_PUBLIC
Automated g4 rollback of changelist 159643278
PiperOrigin-RevId: 160894669
Diffstat (limited to 'third_party/gpus/cuda')
-rw-r--r-- | third_party/gpus/cuda/BUILD.tpl | 82 |
1 files changed, 55 insertions, 27 deletions
diff --git a/third_party/gpus/cuda/BUILD.tpl b/third_party/gpus/cuda/BUILD.tpl index 87c8678714..51d9e4e994 100644 --- a/third_party/gpus/cuda/BUILD.tpl +++ b/third_party/gpus/cuda/BUILD.tpl @@ -40,20 +40,23 @@ config_setting( cc_library( name = "cuda_headers", hdrs = [ - "cuda_config.h", + "cuda/cuda_config.h", %{cuda_headers} ], includes = [ ".", - "include", + "cuda/include", ], visibility = ["//visibility:public"], ) cc_library( name = "cudart_static", - srcs = ["lib/%{cudart_static_lib}"], - includes = ["include"], + srcs = ["cuda/lib/%{cudart_static_lib}"], + includes = [ + ".", + "cuda/include", + ], linkopts = select({ ":freebsd": [], "//conditions:default": ["-ldl"], @@ -66,34 +69,46 @@ cc_library( cc_library( name = "cuda_driver", - srcs = ["lib/%{cuda_driver_lib}"], - includes = ["include"], + srcs = ["cuda/lib/%{cuda_driver_lib}"], + includes = [ + ".", + "cuda/include", + ], visibility = ["//visibility:public"], ) cc_library( name = "cudart", - srcs = ["lib/%{cudart_lib}"], - data = ["lib/%{cudart_lib}"], - includes = ["include"], + srcs = ["cuda/lib/%{cudart_lib}"], + data = ["cuda/lib/%{cudart_lib}"], + includes = [ + ".", + "cuda/include", + ], linkstatic = 1, visibility = ["//visibility:public"], ) cc_library( name = "cublas", - srcs = ["lib/%{cublas_lib}"], - data = ["lib/%{cublas_lib}"], - includes = ["include"], + srcs = ["cuda/lib/%{cublas_lib}"], + data = ["cuda/lib/%{cublas_lib}"], + includes = [ + ".", + "cuda/include", + ], linkstatic = 1, visibility = ["//visibility:public"], ) cc_library( name = "cusolver", - srcs = ["lib/%{cusolver_lib}"], - data = ["lib/%{cusolver_lib}"], - includes = ["include"], + srcs = ["cuda/lib/%{cusolver_lib}"], + data = ["cuda/lib/%{cusolver_lib}"], + includes = [ + ".", + "cuda/include", + ], linkopts = ["-lgomp"], linkstatic = 1, visibility = ["//visibility:public"], @@ -101,27 +116,36 @@ cc_library( cc_library( name = "cudnn", - srcs = ["lib/%{cudnn_lib}"], - data = ["lib/%{cudnn_lib}"], - includes = ["include"], + srcs = ["cuda/lib/%{cudnn_lib}"], + data = ["cuda/lib/%{cudnn_lib}"], + includes = [ + ".", + "cuda/include", + ], linkstatic = 1, visibility = ["//visibility:public"], ) cc_library( name = "cufft", - srcs = ["lib/%{cufft_lib}"], - data = ["lib/%{cufft_lib}"], - includes = ["include"], + srcs = ["cuda/lib/%{cufft_lib}"], + data = ["cuda/lib/%{cufft_lib}"], + includes = [ + ".", + "cuda/include", + ], linkstatic = 1, visibility = ["//visibility:public"], ) cc_library( name = "curand", - srcs = ["lib/%{curand_lib}"], - data = ["lib/%{curand_lib}"], - includes = ["include"], + srcs = ["cuda/lib/%{curand_lib}"], + data = ["cuda/lib/%{curand_lib}"], + includes = [ + ".", + "cuda/include", + ], linkstatic = 1, visibility = ["//visibility:public"], ) @@ -142,19 +166,23 @@ cc_library( cc_library( name = "cupti_headers", hdrs = [ - "cuda_config.h", + "cuda/cuda_config.h", ":cuda-extras", ], includes = [ ".", - "extras/CUPTI/include/", + "cuda/extras/CUPTI/include/", ], visibility = ["//visibility:public"], ) cc_library( name = "cupti_dsos", - data = ["lib/%{cupti_lib}"], + data = ["cuda/lib/%{cupti_lib}"], + includes = [ + ".", + "cuda/include", + ], visibility = ["//visibility:public"], ) |