wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-19 12:30:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix plasma bug (#7322)

Browse Source
This commit is contained in:
Stephanie Wang
2020-02-25 18:15:28 -08:00
committed by GitHub
parent 44b4394afa
commit 9964657815
2 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ray/core_worker/core_worker.cc
+10 -6
View File
@@ -480,16 +480,20 @@ Status CoreWorker::Get(const std::vector<ObjectID> &ids, const int64_t timeout_m
&result_map, &got_exception));
}
// Erase any objects that were promoted to plasma from the results. These get
// requests will be retried at the plasma store.
for (auto it = result_map.begin(); it != result_map.end(); it++) {
if (it->second->IsInPlasmaError()) {
RAY_LOG(DEBUG) << it->first << " in plasma, doing fetch-and-get";
plasma_object_ids.insert(it->first);
result_map.erase(it);
}
}
if (!got_exception) {
// If any of the objects have been promoted to plasma, then we retry their
// gets at the provider plasma. Once we get the objects from plasma, we flip
// the transport type again and return them for the original direct call ids.
for (const auto &pair : result_map) {
if (pair.second->IsInPlasmaError()) {
RAY_LOG(DEBUG) << pair.first << " in plasma, doing fetch-and-get";
plasma_object_ids.insert(pair.first);
}
}
int64_t local_timeout_ms = timeout_ms;
if (timeout_ms >= 0) {
local_timeout_ms = std::max(static_cast<int64_t>(0),