From fc8998416286272e702890613b9dfef99d75cb65 Mon Sep 17 00:00:00 2001
From: Stephanie Wang <swang@cs.berkeley.edu>
Date: Wed, 10 Feb 2021 12:22:08 -0800
Subject: [PATCH] Subtract from num bytes in use (#13944)

---
 src/ray/object_manager/plasma/store.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/ray/object_manager/plasma/store.cc b/src/ray/object_manager/plasma/store.cc
index af7219273..920ced48e 100644
--- a/src/ray/object_manager/plasma/store.cc
+++ b/src/ray/object_manager/plasma/store.cc
@@ -571,6 +571,10 @@ void PlasmaStore::EraseFromObjectTable(const ObjectID &object_id) {
   if (object->device_num == 0) {
     PlasmaAllocator::Free(object->pointer, buff_size);
   }
+  if (object->ref_count > 0) {
+    // A client was using this object.
+    num_bytes_in_use_ -= object->data_size + object->metadata_size;
+  }
   store_info_.objects.erase(object_id);
 }