diff options
-rw-r--r-- | tensorflow/stream_executor/host/host_gpu_executor.cc | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/stream_executor/host/host_gpu_executor.cc b/tensorflow/stream_executor/host/host_gpu_executor.cc index 830bc9a681..259bd10238 100644 --- a/tensorflow/stream_executor/host/host_gpu_executor.cc +++ b/tensorflow/stream_executor/host/host_gpu_executor.cc @@ -19,6 +19,7 @@ limitations under the License. #include <string.h> +#include "tensorflow/core/platform/profile_utils/cpu_utils.h" #include "tensorflow/stream_executor/host/host_platform_id.h" #include "tensorflow/stream_executor/host/host_stream.h" #include "tensorflow/stream_executor/host/host_timer.h" @@ -190,7 +191,10 @@ DeviceDescription *HostExecutor::PopulateDeviceDescription() const { // doesn't result in thrashing or other badness? 4GiB chosen arbitrarily. builder.set_device_memory_size(static_cast<uint64>(4) * 1024 * 1024 * 1024); - builder.set_clock_rate_ghz(static_cast<float>(CLOCKS_PER_SEC) / 1e9); + builder.set_clock_rate_ghz( + static_cast<float>( + tensorflow::profile_utils::CpuUtils::GetCycleCounterFrequency()) / + 1e9); auto built = builder.Build(); return built.release(); |