[core][bugfix] fix print ref count for an erased iterator (ray-projec…

…t#20138)
kaydoh · Nov 9, 2021 · 6bb0659 · 6bb0659
1 parent 39b3eb9
commit 6bb0659
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/ray/core_worker/reference_count.cc b/src/ray/core_worker/reference_count.cc
@@ -277,8 +277,9 @@ void ReferenceCounter::RemoveLocalReference(const ObjectID &object_id,
   PRINT_REF_COUNT(it);
   if (it->second.RefCount() == 0) {
     DeleteReferenceInternal(it, deleted);
+  } else {
+    PRINT_REF_COUNT(it);
   }
-  PRINT_REF_COUNT(it);
 }
 
 void ReferenceCounter::UpdateSubmittedTaskReferences(