| 57 | |
| 58 | template <typename K, typename V> |
| 59 | void CheckpointLoader<K, V>::RestoreSSD() { |
| 60 | std::string name_string_temp(restore_args_.m_name_string); |
| 61 | std::string new_str = "_"; |
| 62 | int64 pos = name_string_temp.find("/"); |
| 63 | while (pos != std::string::npos) { |
| 64 | name_string_temp.replace(pos, 1, new_str.data(), 1); |
| 65 | pos = name_string_temp.find("/"); |
| 66 | } |
| 67 | std::string ssd_record_file_name = restore_args_.m_file_name_string + "-" + |
| 68 | name_string_temp + "-ssd_record"; |
| 69 | if (Env::Default()->FileExists(ssd_record_file_name + ".index").ok()) { |
| 70 | std::string ssd_emb_file_name = restore_args_.m_file_name_string + "-" + |
| 71 | name_string_temp + "-emb_files"; |
| 72 | BundleReader ssd_record_reader(Env::Default(), ssd_record_file_name); |
| 73 | RestoreSSDBuffer<K> ssd_buffer(&ssd_record_reader); |
| 74 | VLOG(1) << "Loading SSD record... " << ssd_record_file_name; |
| 75 | storage_->RestoreSSD(ev_->GetEmbeddingIndex(), |
| 76 | ev_->GetEmbeddingSlotNum(), ev_->ValueLen(), |
| 77 | ssd_emb_file_name, ev_, ssd_buffer); |
| 78 | } |
| 79 | } |
| 80 | #define REGISTER_KERNELS(ktype, vtype) \ |
| 81 | template void CheckpointLoader<ktype, vtype>::RestoreSSD(); |
| 82 | #define REGISTER_KERNELS_ALL_INDEX(type) \ |
nothing calls this directly
no test coverage detected