aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/stream_executor/stream_executor_pimpl.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-08-17 14:56:25 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-08-17 15:03:24 -0700
commitb48cfaea2aea3707a33e60c10385a87e37101b95 (patch)
tree897738a42d520918efcc9bbece881c5c2fcce583 /tensorflow/stream_executor/stream_executor_pimpl.cc
parentd30537a10529a73b712d805a1f26fd39ce5be609 (diff)
Make tensorflow::mutex implement a shared (reader/writer) lock, using
open source nsync library. PiperOrigin-RevId: 165633487
Diffstat (limited to 'tensorflow/stream_executor/stream_executor_pimpl.cc')
-rw-r--r--tensorflow/stream_executor/stream_executor_pimpl.cc6
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/stream_executor/stream_executor_pimpl.cc b/tensorflow/stream_executor/stream_executor_pimpl.cc
index b3eefe0299..4cac6f6c96 100644
--- a/tensorflow/stream_executor/stream_executor_pimpl.cc
+++ b/tensorflow/stream_executor/stream_executor_pimpl.cc
@@ -119,7 +119,7 @@ class ScopedTracer {
void Trace(CallbackT callback, TraceArgsT... args) {
{
// Instance tracers held in a block to limit the lock lifetime.
- shared_lock lock{stream_exec_->mu_};
+ tf_shared_lock lock{stream_exec_->mu_};
for (TraceListener *listener : stream_exec_->listeners_) {
(listener->*callback)(correlation_id_,
std::forward<TraceArgsT>(args)...);
@@ -229,7 +229,7 @@ void StreamExecutor::Deallocate(DeviceMemoryBase *mem) {
}
void StreamExecutor::GetMemAllocs(std::map<void *, AllocRecord> *records_out) {
- shared_lock lock{mu_};
+ tf_shared_lock lock{mu_};
*records_out = mem_allocs_;
}
@@ -754,7 +754,7 @@ void StreamExecutor::SubmitTrace(TraceCallT trace_call, ArgsT &&... args) {
if (tracing_enabled_) {
{
// instance tracers held in a block to limit the lock lifetime.
- shared_lock lock{mu_};
+ tf_shared_lock lock{mu_};
for (TraceListener *listener : listeners_) {
(listener->*trace_call)(std::forward<ArgsT>(args)...);
}