From fc526374abac9e1080e06470004ab292fe0c162a Mon Sep 17 00:00:00 2001
From: Camillo Lugaresi <camillol@google.com>
Date: Mon, 28 Nov 2022 17:48:37 -0800
Subject: [PATCH] Use GpuResources in GpuTestBase and update
 GpuBufferMultiPoolTest

PiperOrigin-RevId: 491486495
---
 mediapipe/gpu/gpu_test_base.h | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mediapipe/gpu/gpu_test_base.h b/mediapipe/gpu/gpu_test_base.h
index e9fd64725..6ec53603b 100644
--- a/mediapipe/gpu/gpu_test_base.h
+++ b/mediapipe/gpu/gpu_test_base.h
@@ -24,13 +24,14 @@ namespace mediapipe {
 
 class GpuTestBase : public ::testing::Test {
  protected:
-  GpuTestBase() { helper_.InitializeForTest(&gpu_shared_); }
+  GpuTestBase() { helper_.InitializeForTest(gpu_resources_.get()); }
 
   void RunInGlContext(std::function<void(void)> gl_func) {
     helper_.RunInGlContext(std::move(gl_func));
   }
 
   GpuSharedData gpu_shared_;
+  std::shared_ptr<GpuResources> gpu_resources_ = gpu_shared_.gpu_resources;
   GlCalculatorHelper helper_;
 };