diff options
Diffstat (limited to 'tensorflow/core/framework/allocator.cc')
| -rw-r--r-- | tensorflow/core/framework/allocator.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/framework/allocator.cc b/tensorflow/core/framework/allocator.cc index 601d87fa55..812ce4bfe7 100644 --- a/tensorflow/core/framework/allocator.cc +++ b/tensorflow/core/framework/allocator.cc @@ -68,7 +68,7 @@ class CPUAllocator : public Allocator { string Name() override { return "cpu"; } void* AllocateRaw(size_t alignment, size_t num_bytes) override { - void* p = port::aligned_malloc(num_bytes, alignment); + void* p = port::AlignedMalloc(num_bytes, alignment); if (cpu_allocator_collect_stats) { const std::size_t alloc_size = port::MallocExtension_GetAllocatedSize(p); mutex_lock l(mu_); @@ -89,7 +89,7 @@ class CPUAllocator : public Allocator { mutex_lock l(mu_); stats_.bytes_in_use -= alloc_size; } - port::aligned_free(ptr); + port::AlignedFree(ptr); } void GetStats(AllocatorStats* stats) override { |