diff --git a/tensorflow/core/platform/default/gpu/BUILD b/tensorflow/core/platform/default/gpu/BUILD index 6b0c919a89a..3965c7d2ec6 100644 --- a/tensorflow/core/platform/default/gpu/BUILD +++ b/tensorflow/core/platform/default/gpu/BUILD @@ -15,7 +15,6 @@ tf_cuda_library( copts = tf_copts(), cuda_deps = [ "//tensorflow/core:stream_executor", - "@local_config_cuda//cuda:cuda_headers", "@local_config_cuda//cuda:cupti_headers", ], data = ["@local_config_cuda//cuda:cupti_dsos"], diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 47a83d51a47..c56231fa1c8 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -46,6 +46,7 @@ load( load( "@local_config_cuda//cuda:build_defs.bzl", "if_cuda", + "cuda_path_flags" ) # List of proto files for android builds @@ -425,7 +426,6 @@ def _cuda_copts(): common_cuda_opts + [ "-fcuda-flush-denormals-to-zero", - "--cuda-path=external/local_config_cuda/cuda", "--cuda-gpu-arch=sm_35", ] ), @@ -434,7 +434,7 @@ def _cuda_copts(): # optimizations are not enabled at O2. "@local_config_cuda//cuda:using_clang_opt": ["-O3"], "//conditions:default": [], - }) + }) + cuda_path_flags() # Build defs for TensorFlow kernels @@ -480,7 +480,10 @@ def tf_cuda_library(deps=None, cuda_deps=None, copts=None, **kwargs): copts = [] native.cc_library( - deps = deps + if_cuda(cuda_deps + ["//tensorflow/core:cuda"]), + deps = deps + if_cuda(cuda_deps + [ + "//tensorflow/core:cuda", + "@local_config_cuda//cuda:cuda_headers" + ]), copts = copts + if_cuda(["-DGOOGLE_CUDA=1"]), **kwargs) diff --git a/third_party/gpus/cuda/build_defs.bzl.tpl b/third_party/gpus/cuda/build_defs.bzl.tpl index 31aaf9ac285..769ad3378e2 100644 --- a/third_party/gpus/cuda/build_defs.bzl.tpl +++ b/third_party/gpus/cuda/build_defs.bzl.tpl @@ -1,4 +1,7 @@ # Macros for building CUDA code. +def cuda_path_flags(): + """Stub for compatibility with internal build.""" + return [] def if_cuda(if_true, if_false = []): """Shorthand for select()'ing on whether we're building with CUDA.