[Embedding] Add return value of RestoreSSD of DramSSDHashStorage.

Signed-off-by: lixy9474 <[email protected]>
DeepRec-AI · Sep 5, 2023 · 24858bf · 24858bf
1 parent 8d8e16a
commit 24858bf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow/core/framework/embedding/dram_ssd_storage.h b/tensorflow/core/framework/embedding/dram_ssd_storage.h
@@ -181,7 +181,9 @@ class DramSsdHashStorage : public MultiTierStorage<K, V> {
                       restore_buff.key_offset_list_buf,
                       restore_buff.num_of_keys,
                       file_id_map);
+    return Status::OK();
   }
+
   Status Eviction(K* evict_ids, int64 evict_size) override {
     ValuePtr<V>* value_ptr = nullptr;
     for (int64 i = 0; i < evict_size; ++i) {