No public description

PiperOrigin-RevId: 608254035
kp-forks · Feb 19, 2024 · 32647d2 · 32647d2
1 parent a620309
commit 32647d2
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/mediapipe/gpu/reusable_pool.h b/mediapipe/gpu/reusable_pool.h
@@ -85,12 +85,8 @@ inline absl::StatusOr<std::shared_ptr<Item>> ReusablePool<Item>::GetBuffer() {
   {
     absl::MutexLock lock(&mutex_);
     if (available_.empty()) {
-      // TODO - propagate absl::Status to GetBuffer caller.
-      auto buffer_from_factory = item_factory_();
-      if (!buffer_from_factory.ok() || *buffer_from_factory == nullptr) {
-        return nullptr;
-      }
-      buffer = *std::move(buffer_from_factory);
+      MP_ASSIGN_OR_RETURN(buffer, item_factory_());
+      RET_CHECK_NE(buffer, nullptr) << "Failed to create buffer";
     } else {
       buffer = std::move(available_.back());
       available_.pop_back();