diff options
author | 2016-08-25 10:22:44 -0800 | |
---|---|---|
committer | 2016-08-25 11:32:33 -0700 | |
commit | 2c598e874e6a7b6b3185846ce9bac97a7d5d0169 (patch) | |
tree | cdbfbf01e56660d24c12dc0ef664f251032642ef /tensorflow/stream_executor/BUILD | |
parent | 1fa09b5560144e8dd16756d7877e01647fa68325 (diff) |
Merge changes from github.
Change: 131310818
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, ) |