MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / dump_to_json

Method dump_to_json

src/plugin/impl/static_mem_record.cpp:11–36  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

9using namespace cg;
10
11void StaticMemRecorder::dump_to_json() {
12 VisableDataSet writer(m_log_dir);
13 for (auto&& i : m_memory_chunk_recorder) {
14 // static mem chunk
15 if (i.id < m_weight_chunk_id) {
16 std::string overwrite_dest_id =
17 i.is_overwrite ? std::to_string(i.overwrite_dest_id) : "-1";
18 Chunk c(std::to_string(i.id), Chunk::static_mem,
19 std::to_string(i.time_begin), std::to_string(i.time_end),
20 std::to_string(i.addr_begin), std::to_string(i.addr_end),
21 overwrite_dest_id);
22 writer.dump_info(c);
23 } else {
24 // weight mem chunk
25 Chunk c(std::to_string(i.id), Chunk::weight_mem,
26 std::to_string(i.time_begin), std::to_string(i.time_end),
27 std::to_string(i.addr_begin), std::to_string(i.addr_end), "-1");
28 writer.dump_info(c);
29 }
30 }
31 for (auto&& i : m_opr_seq_recorder) {
32 OprSeq o(std::to_string(i.id), i.name);
33 writer.dump_info(o);
34 }
35 writer.write_to_file();
36}
37#endif
38#endif

Callers 1

do_registMethod · 0.80

Calls 3

to_stringFunction · 0.85
dump_infoMethod · 0.80
write_to_fileMethod · 0.45

Tested by

no test coverage detected