update

bytedance · Jan 19, 2021 · 83c6c8a · 83c6c8a
1 parent 40ec326
commit 83c6c8a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/byteps/server/server.cc b/byteps/server/server.cc
@@ -56,10 +56,10 @@ void SendPullResponse(const DataHandleType type, const uint64_t key,
     data = updates.merged.tensor;
     len = updates.merged.len;
   } else {
-    auto stored = GetStore(key);
-    CHECK(stored->tensor) << "init " << key << " first";
-    data = stored->tensor;
-    len = stored->len;
+    auto stored = store_.at(key);
+    CHECK(stored.tensor) << "init " << key << " first";
+    data = stored.tensor;
+    len = stored.len;
   }
 
   // send pull response