diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-20 10:08:23 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-20 10:10:54 -0700 |
commit | c023f46956f8a867d0dc77f1ee742564a3622e68 (patch) | |
tree | 7e5dd1ac054e7cdf6e1f859ed4c131a03bf645ba /tensorflow/core/platform | |
parent | 7fc43521ee5dd12525afbc2ad766f562b24f0043 (diff) |
Modify AllocatorRegistry to be an AllocatorFactoryRegistry
in preparation for using NUMA node specific allocators.
Also, add stub NUMA functions on the platform interface to
platform/windows/port.cc.
PiperOrigin-RevId: 205413998
Diffstat (limited to 'tensorflow/core/platform')
-rw-r--r-- | tensorflow/core/platform/windows/port.cc | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/tensorflow/core/platform/windows/port.cc b/tensorflow/core/platform/windows/port.cc index f2aaf13bec..5375f56372 100644 --- a/tensorflow/core/platform/windows/port.cc +++ b/tensorflow/core/platform/windows/port.cc @@ -33,6 +33,7 @@ limitations under the License. #include "tensorflow/core/platform/init_main.h" #include "tensorflow/core/platform/logging.h" #include "tensorflow/core/platform/mem.h" +#include "tensorflow/core/platform/numa.h" #include "tensorflow/core/platform/snappy.h" #include "tensorflow/core/platform/types.h" @@ -57,6 +58,17 @@ int NumSchedulableCPUs() { return system_info.dwNumberOfProcessors; } +bool NUMAEnabled() { + // Not yet implemented: coming soon. + return false; +} + +int NUMANumNodes() { return 1; } + +void NUMASetThreadNodeAffinity(int node) {} + +int NUMAGetThreadNodeAffinity() { return kNUMANoAffinity; } + void* AlignedMalloc(size_t size, int minimum_alignment) { #ifdef TENSORFLOW_USE_JEMALLOC void* ptr = NULL; @@ -108,6 +120,14 @@ void Free(void* ptr) { #endif } +void* NUMAMalloc(int node, size_t size, int minimum_alignment) { + return AlignedMalloc(size, minimum_alignment); +} + +void NUMAFree(void* ptr, size_t size) { Free(ptr); } + +int NUMAGetMemAffinity(const void* addr) { return kNUMANoAffinity; } + void MallocExtension_ReleaseToSystem(std::size_t num_bytes) { // No-op. } |