| 2315 | } |
| 2316 | |
| 2317 | py::dict ExecutorMetaToDict(const ExecutorMetaMap &meta) { |
| 2318 | py::dict d; |
| 2319 | for (const auto &stat : meta) { |
| 2320 | py::dict op_dict; |
| 2321 | py::list real_memory_size; |
| 2322 | py::list reserved_memory_size; |
| 2323 | py::list max_real_memory_size; |
| 2324 | py::list max_reserved_memory_size; |
| 2325 | for (const auto &entry : stat.second) { |
| 2326 | real_memory_size.append(entry.real_size); |
| 2327 | max_real_memory_size.append(entry.max_real_size); |
| 2328 | reserved_memory_size.append(entry.reserved); |
| 2329 | max_reserved_memory_size.append(entry.max_reserved); |
| 2330 | } |
| 2331 | op_dict["real_memory_size"] = real_memory_size; |
| 2332 | op_dict["max_real_memory_size"] = max_real_memory_size; |
| 2333 | op_dict["reserved_memory_size"] = reserved_memory_size; |
| 2334 | op_dict["max_reserved_memory_size"] = max_reserved_memory_size; |
| 2335 | d[stat.first.c_str()] = op_dict; |
| 2336 | } |
| 2337 | return d; |
| 2338 | } |
| 2339 | |
| 2340 | void ExposePipelineDebug(py::module &m) { |
| 2341 | py::class_<PipelineDebug>(m, "PipelineDebug") |
no test coverage detected