diff options
Diffstat (limited to 'tensorflow/stream_executor/BUILD')
-rw-r--r-- | tensorflow/stream_executor/BUILD | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/tensorflow/stream_executor/BUILD b/tensorflow/stream_executor/BUILD index 24e8305d31..256b128750 100644 --- a/tensorflow/stream_executor/BUILD +++ b/tensorflow/stream_executor/BUILD @@ -27,9 +27,10 @@ cc_library( ]), data = [ "//tensorflow/core:cuda", - "//third_party/gpus/cuda:cublas", - "//third_party/gpus/cuda:cudnn", - "//third_party/gpus/cuda:cufft", + "@local_config_cuda//cuda:cublas", + "@local_config_cuda//cuda:cudnn", + "@local_config_cuda//cuda:cufft", + "@local_config_cuda//cuda:curand", ], linkopts = [ "-ldl", @@ -37,7 +38,7 @@ cc_library( visibility = ["//visibility:public"], deps = [ "//tensorflow/core:lib", - "//third_party/gpus/cuda:cuda_headers", + "@local_config_cuda//cuda:cuda_headers", ], alwayslink = 1, ) |