| 178 | } |
| 179 | |
| 180 | void NodeExecStatsWrapper::AddAllocation( |
| 181 | Allocator* allocator, TrackingAllocator* tracking_allocator) { |
| 182 | AllocatorMemoryUsed* memory = stats_->add_memory(); |
| 183 | memory->set_allocator_name(allocator->Name()); |
| 184 | auto sizes = tracking_allocator->GetSizes(); |
| 185 | memory->set_total_bytes(std::get<0>(sizes)); |
| 186 | memory->set_peak_bytes(std::get<1>(sizes)); |
| 187 | memory->set_live_bytes(std::get<2>(sizes)); |
| 188 | |
| 189 | absl::optional<AllocatorStats> stats = allocator->GetStats(); |
| 190 | if (stats) { |
| 191 | memory->set_allocator_bytes_in_use(stats->bytes_in_use); |
| 192 | } |
| 193 | allocations_.push_back(std::make_pair(memory, tracking_allocator)); |
| 194 | } |
| 195 | |
| 196 | void NodeExecStatsWrapper::SetActivityID(int64 id) { |
| 197 | stats_->set_correlation_id(id); |