=================== TensorRTEngineCacheIO ============= */
| 138 | |
| 139 | /* =================== TensorRTEngineCacheIO ============= */ |
| 140 | void TensorRTEngineCacheIO::read_cache() { |
| 141 | uint32_t nr_engines; |
| 142 | read(nr_engines); |
| 143 | for (uint32_t i = 0; i < nr_engines; ++i) { |
| 144 | uint32_t key_size; |
| 145 | read(key_size); |
| 146 | std::string key; |
| 147 | key.resize(key_size); |
| 148 | read(const_cast<char*>(key.data()), key.size()); |
| 149 | mgb_log_debug("read key: %s", key.c_str()); |
| 150 | m_cache[std::move(key)].init_from(*this); |
| 151 | } |
| 152 | } |
| 153 | |
| 154 | TensorRTEngineCacheIO::TensorRTEngineCacheIO(std::string filename) |
| 155 | : m_filename{std::move(filename)} { |