1 /* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
2
3 Licensed under the Apache License, Version 2.0 (the "License");
4 you may not use this file except in compliance with the License.
5 You may obtain a copy of the License at
6
7 http://www.apache.org/licenses/LICENSE-2.0
8
9 Unless required by applicable law or agreed to in writing, software
10 distributed under the License is distributed on an "AS IS" BASIS,
11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 See the License for the specific language governing permissions and
13 limitations under the License.
14 ==============================================================================*/
15
16 #include "tensorflow/lite/delegates/gpu/gl/gl_buffer.h"
17
18 #include <memory>
19
20 #include <gmock/gmock.h>
21 #include <gtest/gtest.h>
22 #include "tensorflow/lite/delegates/gpu/common/status.h"
23 #include "tensorflow/lite/delegates/gpu/gl/egl_environment.h"
24
25 namespace tflite {
26 namespace gpu {
27 namespace gl {
28 namespace {
29
TEST(Buffer,Read)30 TEST(Buffer, Read) {
31 std::unique_ptr<EglEnvironment> env;
32 ASSERT_TRUE(EglEnvironment::NewEglEnvironment(&env).ok());
33 std::vector<float> test = {0, 1, 2, 3};
34 GlBuffer buffer;
35 ASSERT_TRUE(CreateReadOnlyShaderStorageBuffer<float>(test, &buffer).ok());
36 std::vector<float> from_buffer;
37 ASSERT_TRUE(AppendFromBuffer(buffer, &from_buffer).ok());
38 EXPECT_EQ(test, from_buffer);
39 }
40
TEST(Buffer,Write)41 TEST(Buffer, Write) {
42 std::unique_ptr<EglEnvironment> env;
43 ASSERT_TRUE(EglEnvironment::NewEglEnvironment(&env).ok());
44 GlBuffer buffer;
45 ASSERT_TRUE(CreateReadWriteShaderStorageBuffer<float>(4, &buffer).ok());
46 std::vector<float> test = {0, 1, 2, 3};
47 ASSERT_TRUE(buffer.Write<float>(test).ok());
48 std::vector<float> from_buffer;
49 ASSERT_TRUE(AppendFromBuffer(buffer, &from_buffer).ok());
50 EXPECT_EQ(test, from_buffer);
51 }
52
TEST(Buffer,View)53 TEST(Buffer, View) {
54 std::unique_ptr<EglEnvironment> env;
55 ASSERT_TRUE(EglEnvironment::NewEglEnvironment(&env).ok());
56 GlBuffer buffer;
57 ASSERT_TRUE(CreateReadWriteShaderStorageBuffer<float>(6, &buffer).ok());
58 EXPECT_TRUE(buffer.has_ownership());
59 EXPECT_EQ(24, buffer.bytes_size());
60 EXPECT_EQ(0, buffer.offset());
61
62 // Create view and write data there.
63 GlBuffer view;
64 ASSERT_TRUE(buffer.MakeView(4, 16, &view).ok());
65 EXPECT_FALSE(view.has_ownership());
66 EXPECT_EQ(16, view.bytes_size());
67 EXPECT_EQ(4, view.offset());
68 std::vector<float> test = {1, 2, 3, 4};
69 ASSERT_TRUE(view.Write<float>(test).ok());
70
71 // Check that data indeed landed in a buffer with proper offset.
72 std::vector<float> from_buffer;
73 ASSERT_TRUE(AppendFromBuffer(buffer, &from_buffer).ok());
74 EXPECT_THAT(from_buffer, testing::ElementsAre(0, 1, 2, 3, 4, 0));
75
76 std::vector<float> from_view;
77 ASSERT_TRUE(AppendFromBuffer(view, &from_view).ok());
78 EXPECT_THAT(from_view, testing::ElementsAre(1, 2, 3, 4));
79 }
80
TEST(Buffer,SubView)81 TEST(Buffer, SubView) {
82 std::unique_ptr<EglEnvironment> env;
83 ASSERT_TRUE(EglEnvironment::NewEglEnvironment(&env).ok());
84 GlBuffer buffer;
85 ASSERT_TRUE(CreateReadWriteShaderStorageBuffer<float>(6, &buffer).ok());
86
87 // Create view and another view over that view.
88
89 GlBuffer view1;
90 ASSERT_TRUE(buffer.MakeView(4, 16, &view1).ok());
91 GlBuffer view2;
92 EXPECT_FALSE(view1.MakeView(1, 16, &view2).ok());
93 ASSERT_TRUE(view1.MakeView(2, 2, &view2).ok());
94
95 EXPECT_FALSE(view2.has_ownership());
96 EXPECT_EQ(2, view2.bytes_size());
97 EXPECT_EQ(6, view2.offset());
98 }
99
TEST(Buffer,Copy)100 TEST(Buffer, Copy) {
101 std::unique_ptr<EglEnvironment> env;
102 ASSERT_TRUE(EglEnvironment::NewEglEnvironment(&env).ok());
103 GlBuffer buffer;
104 ASSERT_TRUE(CreateReadWriteShaderStorageBuffer<float>(4, &buffer).ok());
105
106 // Create view and write data there.
107 GlBuffer view1;
108 ASSERT_TRUE(buffer.MakeView(4, 4, &view1).ok());
109
110 GlBuffer view2;
111 ASSERT_TRUE(buffer.MakeView(8, 4, &view2).ok());
112
113 // Copy data from one view to another
114 ASSERT_TRUE(view1.Write<float>({1}).ok());
115 ASSERT_TRUE(CopyBuffer(view1, view2).ok());
116
117 // Check that data indeed landed correctly.
118 std::vector<float> from_buffer;
119 ASSERT_TRUE(AppendFromBuffer(buffer, &from_buffer).ok());
120 EXPECT_THAT(from_buffer, testing::ElementsAre(0, 1, 1, 0));
121 }
122
123 } // namespace
124 } // namespace gl
125 } // namespace gpu
126 } // namespace tflite
127