From 24858bf704377b46d6d2e309507688440d976b50 Mon Sep 17 00:00:00 2001
From: lixy9474 <lxy268263@alibaba-inc.com>
Date: Tue, 5 Sep 2023 17:17:30 +0800
Subject: [PATCH] [Embedding] Add return value of RestoreSSD of
 DramSSDHashStorage.

Signed-off-by: lixy9474 <lxy268263@alibaba-inc.com>
---
 tensorflow/core/framework/embedding/dram_ssd_storage.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tensorflow/core/framework/embedding/dram_ssd_storage.h b/tensorflow/core/framework/embedding/dram_ssd_storage.h
index 4243cc14eb3..356a61d865f 100644
--- 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) {