(args)
| 184 | |
| 185 | |
| 186 | def convert(args): |
| 187 | file_process_order = { |
| 188 | "graph.json": comp_graph_plotter, |
| 189 | "StaticMemoryInfo.json": peak_mem_regist, |
| 190 | } |
| 191 | g = os.walk(args.input) |
| 192 | for path, dir_list, file_list in g: |
| 193 | out_path = path.replace(args.input, args.output) |
| 194 | writer = SummaryWriterExtend(out_path) |
| 195 | for key, value in file_process_order.items(): |
| 196 | if key in file_list: |
| 197 | value(os.path.join(path, key), writer) |
| 198 | |
| 199 | |
| 200 | def main(): |