Project import generated by Copybara.

GitOrigin-RevId: 6e5aa035cd1f6a9333962df5d3ab97a05bd5744e
2022-06-28 12:11:05 +00:00
parent 4a20e9909d
commit c688862570
144 changed files with 5772 additions and 2118 deletions
@@ -0,0 +1,59 @@
+#include "mediapipe/gpu/gpu_test_base.h"
+#include "testing/base/public/gmock.h"
+#include "testing/base/public/gunit.h"
+
+#ifdef MEDIAPIPE_TENSOR_USE_AHWB
+#include <android/hardware_buffer.h>
+
+#include "mediapipe/framework/formats/tensor.h"
+
+namespace mediapipe {
+
+#if !MEDIAPIPE_DISABLE_GPU
+class TensorAhwbTest : public mediapipe::GpuTestBase {
+ public:
+};
+
+TEST_F(TensorAhwbTest, TestCpuThenAHWB) {
+  Tensor tensor(Tensor::ElementType::kFloat32, Tensor::Shape{1});
+  {
+    auto ptr = tensor.GetCpuWriteView().buffer<float>();
+    EXPECT_NE(ptr, nullptr);
+  }
+  {
+    auto ahwb = tensor.GetAHardwareBufferReadView().handle();
+    EXPECT_NE(ahwb, nullptr);
+  }
+}
+
+TEST_F(TensorAhwbTest, TestAHWBThenCpu) {
+  Tensor tensor(Tensor::ElementType::kFloat32, Tensor::Shape{1});
+  {
+    auto ahwb = tensor.GetAHardwareBufferWriteView().handle();
+    EXPECT_NE(ahwb, nullptr);
+  }
+  {
+    auto ptr = tensor.GetCpuReadView().buffer<float>();
+    EXPECT_NE(ptr, nullptr);
+  }
+}
+
+TEST_F(TensorAhwbTest, TestCpuThenGl) {
+  RunInGlContext([] {
+    Tensor tensor(Tensor::ElementType::kFloat32, Tensor::Shape{1});
+    {
+      auto ptr = tensor.GetCpuWriteView().buffer<float>();
+      EXPECT_NE(ptr, nullptr);
+    }
+    {
+      auto ssbo = tensor.GetOpenGlBufferReadView().name();
+      EXPECT_GT(ssbo, 0);
+    }
+  });
+}
+
+}  // namespace mediapipe
+
+#endif  // !MEDIAPIPE_DISABLE_GPU
+
+#endif  // MEDIAPIPE_TENSOR_USE_AHWB