diff options
author | Peter Hawkins <phawkins@google.com> | 2016-12-05 06:27:11 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-12-05 06:44:09 -0800 |
commit | e4a4e922ed4be890f4273a7a276768dfefe80a4c (patch) | |
tree | b724c791eb601946961899a621ea4335babfd1e0 /tensorflow/stream_executor/host/host_timer.cc | |
parent | b00e490c3834d72aecce445f72b3a717b068d9db (diff) |
Add a Host platform to the stream executor library that runs code on the CPU.
Avoid using a linker-initialized mutex in multi_platform_manager to work around a initialization/destruction ordering problem on Mac.
Include the CUDA platform ID module even in non-CUDA builds.
Change: 141045549
Diffstat (limited to 'tensorflow/stream_executor/host/host_timer.cc')
-rw-r--r-- | tensorflow/stream_executor/host/host_timer.cc | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/host/host_timer.cc b/tensorflow/stream_executor/host/host_timer.cc new file mode 100644 index 0000000000..187db9f0c2 --- /dev/null +++ b/tensorflow/stream_executor/host/host_timer.cc @@ -0,0 +1,52 @@ +/* Copyright 2016 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +==============================================================================*/ + +#include "tensorflow/stream_executor/host/host_timer.h" + +#include "tensorflow/stream_executor/platform/logging.h" +#include "tensorflow/stream_executor/stream.h" +#include "tensorflow/stream_executor/stream_executor_internal.h" + +namespace gpu = ::perftools::gputools; + +namespace perftools { +namespace gputools { +namespace host { + +using std::chrono::duration_cast; + +bool HostTimer::Start(Stream* stream) { + return stream->ThenDoHostCallback([this]() { this->StartNow(); }).ok(); +} + +bool HostTimer::Stop(Stream* stream) { + return stream->ThenDoHostCallback([this]() { this->StopNow(); }).ok(); +} + +uint64 HostTimer::Microseconds() const { + return duration_cast<std::chrono::microseconds>(duration_).count(); +} + +uint64 HostTimer::Nanoseconds() const { + return duration_cast<std::chrono::nanoseconds>(duration_).count(); +} + +void HostTimer::StartNow() { start_time_ = clock::now(); } + +void HostTimer::StopNow() { duration_ = clock::now() - start_time_; } + +} // namespace host +} // namespace gputools +} // namespace perftools |