experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix Windows GPU build

Browse Source 
Rename cublas_virtual_headers to cublas_headers

Fix linking error by exporting commonly used absel symbols

PiperOrigin-RevId: 249426736
This commit is contained in:
A. Unique TensorFlower 2019-05-22 05:01:17 -07:00 committed by TensorFlower Gardener
parent f2784b679d
commit dd6f1a8ad5
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/tools/def_file_filter/def_file_filter.py.tpl
View File

@ -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|"

2
third_party/gpus/cuda/BUILD.windows.tpl vendored
View File

@ -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"],