| 271 | } |
| 272 | |
| 273 | void BasicCPUAllocator::Free(void* ptr, size_t num_bytes) { |
| 274 | if (num_bytes > 0) { |
| 275 | VisitFree(ptr, numa_node_, num_bytes); |
| 276 | if (numa_node_ == port::kNUMANoAffinity) { |
| 277 | port::AlignedFree(ptr); |
| 278 | } else { |
| 279 | port::NUMAFree(ptr, num_bytes); |
| 280 | } |
| 281 | } |
| 282 | } |
| 283 | } // namespace tensorflow |
no test coverage detected