[Embedding] Fix missing return value of RestoreSSD of DramSSDHashStor…

…age. (#926) Signed-off-by: lixy9474 <lxy268263@alibaba-inc.com>
DeepRec-AI · Sep 20, 2023 · 29ecde4 · 29ecde4
1 parent fe194b0
commit 29ecde4
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) {