diff options
Diffstat (limited to 'third_party/gpus/cuda/BUILD.tpl')
-rw-r--r-- | third_party/gpus/cuda/BUILD.tpl | 52 |
1 files changed, 12 insertions, 40 deletions
diff --git a/third_party/gpus/cuda/BUILD.tpl b/third_party/gpus/cuda/BUILD.tpl index d3c307a6a0..20920f7fca 100644 --- a/third_party/gpus/cuda/BUILD.tpl +++ b/third_party/gpus/cuda/BUILD.tpl @@ -1,9 +1,5 @@ licenses(["restricted"]) # MPL2, portions GPL v3, LGPL v3, BSD-like -load("@local_config_cuda//cuda:platform.bzl", "cuda_library_path") -load("@local_config_cuda//cuda:platform.bzl", "cuda_static_library_path") -load("@local_config_cuda//cuda:platform.bzl", "cudnn_library_path") -load("@local_config_cuda//cuda:platform.bzl", "cupti_library_path") load("@local_config_cuda//cuda:platform.bzl", "readlink_command") package(default_visibility = ["//visibility:public"]) @@ -51,9 +47,7 @@ cc_library( cc_library( name = "cudart_static", - srcs = [ - cuda_static_library_path("cudart"), - ], + srcs = ["lib/%{cudart_static_lib}"], includes = ["include/"], linkopts = [ "-ldl", @@ -65,12 +59,8 @@ cc_library( cc_library( name = "cudart", - srcs = [ - cuda_library_path("cudart"), - ], - data = [ - cuda_library_path("cudart"), - ], + srcs = ["lib/%{cudart_lib}"], + data = ["lib/%{cudart_lib}"], includes = ["include/"], linkstatic = 1, visibility = ["//visibility:public"], @@ -78,12 +68,8 @@ cc_library( cc_library( name = "cublas", - srcs = [ - cuda_library_path("cublas"), - ], - data = [ - cuda_library_path("cublas"), - ], + srcs = ["lib/%{cublas_lib}"], + data = ["lib/%{cublas_lib}"], includes = ["include/"], linkstatic = 1, visibility = ["//visibility:public"], @@ -91,12 +77,8 @@ cc_library( cc_library( name = "cudnn", - srcs = [ - cudnn_library_path(), - ], - data = [ - cudnn_library_path(), - ], + srcs = ["lib/%{cudnn_lib}"], + data = ["lib/%{cudnn_lib}"], includes = ["include/"], linkstatic = 1, visibility = ["//visibility:public"], @@ -104,12 +86,8 @@ cc_library( cc_library( name = "cufft", - srcs = [ - cuda_library_path("cufft"), - ], - data = [ - cuda_library_path("cufft"), - ], + srcs = ["lib/%{cufft_lib}"], + data = ["lib/%{cufft_lib}"], includes = ["include/"], linkstatic = 1, visibility = ["//visibility:public"], @@ -117,12 +95,8 @@ cc_library( cc_library( name = "curand", - srcs = [ - cuda_library_path("curand"), - ], - data = [ - cuda_library_path("curand"), - ], + srcs = ["lib/%{curand_lib}"], + data = ["lib/%{curand_lib}"], includes = ["include/"], linkstatic = 1, visibility = ["//visibility:public"], @@ -155,9 +129,7 @@ cc_library( cc_library( name = "cupti_dsos", - data = [ - cupti_library_path(), - ], + data = ["lib/%{cupti_lib}"], visibility = ["//visibility:public"], ) |