| 1166 | } |
| 1167 | |
| 1168 | const std::unordered_map<string, int64> VirtualScheduler::GetPeakMemoryUsage() |
| 1169 | const { |
| 1170 | std::unordered_map<string, int64> result; |
| 1171 | for (const auto& device : device_) { |
| 1172 | const string& name = device.first; |
| 1173 | const DeviceState& state = device.second; |
| 1174 | result[name] = state.max_memory_usage; |
| 1175 | } |
| 1176 | return result; |
| 1177 | } |
| 1178 | |
| 1179 | const std::unordered_map<string, int64> |
| 1180 | VirtualScheduler::GetPersistentMemoryUsage() const { |