diff options
Diffstat (limited to 'tensorflow/core/platform/default/gpu/BUILD')
-rw-r--r-- | tensorflow/core/platform/default/gpu/BUILD | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/core/platform/default/gpu/BUILD b/tensorflow/core/platform/default/gpu/BUILD index 93b6227848..6b0c919a89 100644 --- a/tensorflow/core/platform/default/gpu/BUILD +++ b/tensorflow/core/platform/default/gpu/BUILD @@ -15,9 +15,9 @@ tf_cuda_library( copts = tf_copts(), cuda_deps = [ "//tensorflow/core:stream_executor", - "//third_party/gpus/cuda:cuda_headers", - "//third_party/gpus/cuda:cupti_headers", + "@local_config_cuda//cuda:cuda_headers", + "@local_config_cuda//cuda:cupti_headers", ], - data = ["//third_party/gpus/cuda:cupti_dsos"], + data = ["@local_config_cuda//cuda:cupti_dsos"], visibility = ["//visibility:public"], ) |