From dd6f1a8ad59e9b3ae5d26e227043478cdfb64099 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 22 May 2019 05:01:17 -0700 Subject: [PATCH] Fix Windows GPU build Rename cublas_virtual_headers to cublas_headers Fix linking error by exporting commonly used absel symbols PiperOrigin-RevId: 249426736 --- tensorflow/tools/def_file_filter/def_file_filter.py.tpl | 2 ++ third_party/gpus/cuda/BUILD.windows.tpl | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/tensorflow/tools/def_file_filter/def_file_filter.py.tpl b/tensorflow/tools/def_file_filter/def_file_filter.py.tpl index fda6b86fc2c..329a9bb94ec 100644 --- a/tensorflow/tools/def_file_filter/def_file_filter.py.tpl +++ b/tensorflow/tools/def_file_filter/def_file_filter.py.tpl @@ -50,6 +50,8 @@ INCLUDEPRE_RE = re.compile(r"google::protobuf::internal::ExplicitlyConstructed|" r"google::protobuf::internal::ArenaImpl::AddCleanup|" # for contrib/data/_prefetching_ops r"google::protobuf::internal::LogMessage|" # for contrib/data/_prefetching_ops r"google::protobuf::Arena::OnArenaAllocation|" # for contrib/data/_prefetching_ops + r"absl::Mutex::ReaderLock|" # for //tensorflow/contrib/rnn:python/ops/_gru_ops.so and more ops + r"absl::Mutex::ReaderUnlock|" # for //tensorflow/contrib/rnn:python/ops/_gru_ops.so and more ops r"tensorflow::internal::LogMessage|" r"tensorflow::internal::LogString|" r"tensorflow::internal::CheckOpMessageBuilder|" diff --git a/third_party/gpus/cuda/BUILD.windows.tpl b/third_party/gpus/cuda/BUILD.windows.tpl index ac3b8653a4c..f4bc777568e 100644 --- a/third_party/gpus/cuda/BUILD.windows.tpl +++ b/third_party/gpus/cuda/BUILD.windows.tpl @@ -76,7 +76,7 @@ cc_import( ) cuda_header_library( - name = "cublas_virtual_headers", + name = "cublas_headers", hdrs = [":cublas-include"], include_prefix = "third_party/gpus/cuda/include", includes = ["cublas/include"],