Skip to content

Commit 29ecde4

Browse files
authored
[Embedding] Fix missing return value of RestoreSSD of DramSSDHashStorage. (#926)
Signed-off-by: lixy9474 <[email protected]>
1 parent fe194b0 commit 29ecde4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tensorflow/core/framework/embedding/dram_ssd_storage.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,9 @@ class DramSsdHashStorage : public MultiTierStorage<K, V> {
181181
restore_buff.key_offset_list_buf,
182182
restore_buff.num_of_keys,
183183
file_id_map);
184+
return Status::OK();
184185
}
186+
185187
Status Eviction(K* evict_ids, int64 evict_size) override {
186188
ValuePtr<V>* value_ptr = nullptr;
187189
for (int64 i = 0; i < evict_size; ++i) {

0 commit comments

Comments
 (0)