diff options
Diffstat (limited to 'tensorflow/core/platform/windows/port.cc')
| -rw-r--r-- | tensorflow/core/platform/windows/port.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/platform/windows/port.cc b/tensorflow/core/platform/windows/port.cc index 174f41a993..f2aaf13bec 100644 --- a/tensorflow/core/platform/windows/port.cc +++ b/tensorflow/core/platform/windows/port.cc @@ -171,5 +171,10 @@ int64 AvailableRam() { return INT64_MAX; } +int NumHyperthreadsPerCore() { + static const int ht_per_core = tensorflow::port::CPUIDNumSMT(); + return (ht_per_core > 0) ? ht_per_core : 1; +} + } // namespace port } // namespace tensorflow |