diff options
author | Rohan Jain <rohanj@google.com> | 2018-01-26 11:23:02 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-26 11:27:23 -0800 |
commit | c99daaf104105a4e711d91dca8c73c1badecbe5f (patch) | |
tree | 632871f9669659373ccbc06cd1e8d4df910d127e /tensorflow/stream_executor/multi_platform_manager.cc | |
parent | a0a524e2b6e0e34ab79595a1cba80f5357ee8724 (diff) |
Performance improvements to some GPU code to use shared locks instead of unique locks for some hotspot cases.
PiperOrigin-RevId: 183418559
Diffstat (limited to 'tensorflow/stream_executor/multi_platform_manager.cc')
-rw-r--r-- | tensorflow/stream_executor/multi_platform_manager.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/stream_executor/multi_platform_manager.cc b/tensorflow/stream_executor/multi_platform_manager.cc index cc32a6beaa..f23224ae77 100644 --- a/tensorflow/stream_executor/multi_platform_manager.cc +++ b/tensorflow/stream_executor/multi_platform_manager.cc @@ -45,7 +45,7 @@ namespace gputools { /* static */ port::StatusOr<Platform*> MultiPlatformManager::PlatformWithName( const string& target) { - mutex_lock lock(GetPlatformsMutex()); + tf_shared_lock lock(GetPlatformsMutex()); auto it = GetPlatformMap()->find(port::Lowercase(target)); if (it == GetPlatformMap()->end()) { @@ -59,7 +59,7 @@ namespace gputools { /* static */ port::StatusOr<Platform*> MultiPlatformManager::PlatformWithId( const Platform::Id& id) { - mutex_lock lock(GetPlatformsMutex()); + tf_shared_lock lock(GetPlatformsMutex()); auto it = GetPlatformByIdMap()->find(id); if (it == GetPlatformByIdMap()->end()) { return port::Status( |