blob: c2ecd9aed60a901446beeac83651d1da96918f83 [file] [log] [blame]
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -07001/*
2 * Copyright (C) 2017 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
Michael Butlercf22a572017-09-22 13:26:12 -070017#include "Callbacks.h"
Michael Butler814d8372019-01-15 11:02:55 -080018#include "ExecutionBurstController.h"
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -070019#include "TestHarness.h"
Miao Wanga2d04c82018-02-05 17:26:54 -080020#include "Utils.h"
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -070021
22#include <android-base/logging.h>
Miao Wanga2d04c82018-02-05 17:26:54 -080023#include <android/hardware/neuralnetworks/1.0/IDevice.h>
24#include <android/hardware/neuralnetworks/1.0/IExecutionCallback.h>
25#include <android/hardware/neuralnetworks/1.0/IPreparedModel.h>
26#include <android/hardware/neuralnetworks/1.0/IPreparedModelCallback.h>
27#include <android/hardware/neuralnetworks/1.0/types.h>
Xusong Wangb5cb8f72018-10-31 08:43:12 -070028#include <android/hardware/neuralnetworks/1.1/IDevice.h>
29#include <android/hardware/neuralnetworks/1.2/IDevice.h>
30#include <android/hardware/neuralnetworks/1.2/IExecutionCallback.h>
31#include <android/hardware/neuralnetworks/1.2/IPreparedModel.h>
32#include <android/hardware/neuralnetworks/1.2/IPreparedModelCallback.h>
Miao Wanga2d04c82018-02-05 17:26:54 -080033#include <android/hidl/allocator/1.0/IAllocator.h>
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -070034#include <android/hidl/memory/1.0/IMemory.h>
35#include <hidlmemory/mapping.h>
Michael Butler0897ab32017-10-04 02:38:42 -070036#include <iostream>
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -070037
38namespace android {
39namespace hardware {
40namespace neuralnetworks {
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -070041
42namespace generated_tests {
Xusong Wangb5cb8f72018-10-31 08:43:12 -070043using ::android::hardware::neuralnetworks::V1_2::implementation::ExecutionCallback;
44using ::android::hardware::neuralnetworks::V1_2::implementation::PreparedModelCallback;
Slava Shklyaev9e3fad12018-11-30 17:55:12 +000045using ::test_helper::bool8;
Michael K. Sanders941d61a2018-10-19 14:39:09 +010046using ::test_helper::compare;
47using ::test_helper::expectMultinomialDistributionWithinTolerance;
Mika Raentode166942018-04-17 16:49:50 +010048using ::test_helper::filter;
49using ::test_helper::for_all;
50using ::test_helper::for_each;
Michael K. Sanders941d61a2018-10-19 14:39:09 +010051using ::test_helper::MixedTyped;
52using ::test_helper::MixedTypedExample;
Michael K. Sanders941d61a2018-10-19 14:39:09 +010053using ::test_helper::resize_accordingly;
I-Jui (Ray) Sungf6b85502017-09-20 13:45:50 -070054
I-Jui (Ray) Sung5bf4edf2017-10-06 13:22:39 -070055template <typename T>
Xusong Wanga3165812018-11-19 18:26:08 -080056void copy_back_(std::map<int, std::vector<T>>* dst, const std::vector<RequestArgument>& ra,
57 char* src) {
58 for_each<T>(*dst, [&ra, src](int index, std::vector<T>& m) {
I-Jui (Ray) Sung5bf4edf2017-10-06 13:22:39 -070059 ASSERT_EQ(m.size(), ra[index].location.length / sizeof(T));
I-Jui (Ray) Sungf6b85502017-09-20 13:45:50 -070060 char* begin = src + ra[index].location.offset;
61 memcpy(m.data(), begin, ra[index].location.length);
62 });
63}
64
65void copy_back(MixedTyped* dst, const std::vector<RequestArgument>& ra, char* src) {
Xusong Wanga3165812018-11-19 18:26:08 -080066 copy_back_(&dst->float32Operands, ra, src);
67 copy_back_(&dst->int32Operands, ra, src);
Xusong Wangd49f6652019-01-16 18:32:24 -080068 copy_back_(&dst->quant8AsymmOperands, ra, src);
69 copy_back_(&dst->quant16SymmOperands, ra, src);
Xusong Wanga3165812018-11-19 18:26:08 -080070 copy_back_(&dst->float16Operands, ra, src);
71 copy_back_(&dst->bool8Operands, ra, src);
72 copy_back_(&dst->quant8ChannelOperands, ra, src);
Xusong Wangd49f6652019-01-16 18:32:24 -080073 copy_back_(&dst->quant16AsymmOperands, ra, src);
74 static_assert(8 == MixedTyped::kNumTypes,
Lev Proleev9b490f42018-11-02 12:44:11 +000075 "Number of types in MixedTyped changed, but copy_back function wasn't updated");
I-Jui (Ray) Sungf6b85502017-09-20 13:45:50 -070076}
77
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -070078// Top level driver for models and examples generated by test_generator.py
79// Test driver for those generated from ml/nn/runtime/test/spec
Xusong Wangb5cb8f72018-10-31 08:43:12 -070080static Return<ErrorStatus> ExecutePreparedModel(sp<V1_0::IPreparedModel>& preparedModel,
David Grosse3013492019-01-23 14:01:52 -080081 const Request& request, MeasureTiming,
Xusong Wangb5cb8f72018-10-31 08:43:12 -070082 sp<ExecutionCallback>& callback) {
83 return preparedModel->execute(request, callback);
84}
85static Return<ErrorStatus> ExecutePreparedModel(sp<V1_2::IPreparedModel>& preparedModel,
David Grosse3013492019-01-23 14:01:52 -080086 const Request& request, MeasureTiming measure,
Xusong Wangb5cb8f72018-10-31 08:43:12 -070087 sp<ExecutionCallback>& callback) {
David Grosse3013492019-01-23 14:01:52 -080088 return preparedModel->execute_1_2(request, measure, callback);
Xusong Wangb5cb8f72018-10-31 08:43:12 -070089}
Xusong Wang187c5972018-11-07 09:33:59 -080090static Return<ErrorStatus> ExecutePreparedModel(sp<V1_0::IPreparedModel>&, const Request&,
David Grosse3013492019-01-23 14:01:52 -080091 MeasureTiming, hidl_vec<OutputShape>*, Timing*) {
David Gross49e41672018-12-21 11:20:26 -080092 ADD_FAILURE() << "asking for synchronous execution at V1_0";
93 return ErrorStatus::GENERAL_FAILURE;
94}
95static Return<ErrorStatus> ExecutePreparedModel(sp<V1_2::IPreparedModel>& preparedModel,
David Grosse3013492019-01-23 14:01:52 -080096 const Request& request, MeasureTiming measure,
97 hidl_vec<OutputShape>* outputShapes,
98 Timing* timing) {
Xusong Wang187c5972018-11-07 09:33:59 -080099 ErrorStatus result;
100 Return<void> ret = preparedModel->executeSynchronously(
David Grosse3013492019-01-23 14:01:52 -0800101 request, measure,
102 [&result, outputShapes, timing](ErrorStatus error, const hidl_vec<OutputShape>& shapes,
103 const Timing& time) {
104 result = error;
105 *outputShapes = shapes;
106 *timing = time;
107 });
Xusong Wang187c5972018-11-07 09:33:59 -0800108 if (!ret.isOk()) {
109 return ErrorStatus::GENERAL_FAILURE;
110 }
111 return result;
David Gross49e41672018-12-21 11:20:26 -0800112}
Michael Butler814d8372019-01-15 11:02:55 -0800113static std::unique_ptr<::android::nn::ExecutionBurstController> CreateBurst(
114 const sp<V1_0::IPreparedModel>&) {
115 ADD_FAILURE() << "asking for burst execution at V1_0";
116 return nullptr;
117}
118static std::unique_ptr<::android::nn::ExecutionBurstController> CreateBurst(
119 const sp<V1_2::IPreparedModel>& preparedModel) {
120 return ::android::nn::createExecutionBurstController(preparedModel, /*blocking=*/true);
121}
122enum class Executor { ASYNC, SYNC, BURST };
Xusong Wang929fd212019-01-27 23:08:12 -0800123enum class OutputType { FULLY_SPECIFIED, UNSPECIFIED, INSUFFICIENT };
David Gross49e41672018-12-21 11:20:26 -0800124const float kDefaultAtol = 1e-5f;
125const float kDefaultRtol = 1e-5f;
Xusong Wangb5cb8f72018-10-31 08:43:12 -0700126template <typename T_IPreparedModel>
127void EvaluatePreparedModel(sp<T_IPreparedModel>& preparedModel, std::function<bool(int)> is_ignored,
Michael K. Sandersefa4c812018-10-30 14:44:48 +0000128 const std::vector<MixedTypedExample>& examples,
David Grosse3013492019-01-23 14:01:52 -0800129 bool hasRelaxedFloat32Model, float fpAtol, float fpRtol,
Xusong Wang929fd212019-01-27 23:08:12 -0800130 Executor executor, MeasureTiming measure, OutputType outputType) {
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700131 const uint32_t INPUT = 0;
132 const uint32_t OUTPUT = 1;
133
134 int example_no = 1;
135 for (auto& example : examples) {
136 SCOPED_TRACE(example_no++);
Michael K. Sanders941d61a2018-10-19 14:39:09 +0100137 const MixedTyped& inputs = example.operands.first;
138 const MixedTyped& golden = example.operands.second;
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700139
Xusong Wanga3165812018-11-19 18:26:08 -0800140 const bool hasFloat16Inputs = !inputs.float16Operands.empty();
Michael K. Sandersefa4c812018-10-30 14:44:48 +0000141 if (hasRelaxedFloat32Model || hasFloat16Inputs) {
142 // TODO: Adjust the error limit based on testing.
143 // If in relaxed mode, set the absolute tolerance to be 5ULP of FP16.
144 fpAtol = 5.0f * 0.0009765625f;
145 // Set the relative tolerance to be 5ULP of the corresponding FP precision.
146 fpRtol = 5.0f * 0.0009765625f;
147 }
148
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700149 std::vector<RequestArgument> inputs_info, outputs_info;
150 uint32_t inputSize = 0, outputSize = 0;
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700151 // This function only partially specifies the metadata (vector of RequestArguments).
152 // The contents are copied over below.
153 for_all(inputs, [&inputs_info, &inputSize](int index, auto, auto s) {
154 if (inputs_info.size() <= static_cast<size_t>(index)) inputs_info.resize(index + 1);
155 RequestArgument arg = {
156 .location = {.poolIndex = INPUT, .offset = 0, .length = static_cast<uint32_t>(s)},
157 .dimensions = {},
158 };
I-Jui (Ray) Sung959cd782017-10-04 20:49:57 -0700159 RequestArgument arg_empty = {
160 .hasNoValue = true,
161 };
162 inputs_info[index] = s ? arg : arg_empty;
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700163 inputSize += s;
164 });
165 // Compute offset for inputs 1 and so on
166 {
167 size_t offset = 0;
168 for (auto& i : inputs_info) {
I-Jui (Ray) Sung959cd782017-10-04 20:49:57 -0700169 if (!i.hasNoValue) i.location.offset = offset;
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700170 offset += i.location.length;
171 }
172 }
173
174 MixedTyped test; // holding test results
175
176 // Go through all outputs, initialize RequestArgument descriptors
I-Jui (Ray) Sungf6b85502017-09-20 13:45:50 -0700177 resize_accordingly(golden, test);
Xusong Wang929fd212019-01-27 23:08:12 -0800178 bool sizeLargerThanOne = true;
179 for_all(golden, [&outputs_info, &outputSize, &outputType, &sizeLargerThanOne](
180 int index, auto, auto s) {
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700181 if (outputs_info.size() <= static_cast<size_t>(index)) outputs_info.resize(index + 1);
Xusong Wang929fd212019-01-27 23:08:12 -0800182 if (index == 0) {
183 // On OutputType::INSUFFICIENT, set the output operand with index 0 with
184 // buffer size one byte less than needed.
185 if (outputType == OutputType::INSUFFICIENT) {
186 if (s > 1)
187 s -= 1;
188 else
189 sizeLargerThanOne = false;
190 }
191 }
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700192 RequestArgument arg = {
193 .location = {.poolIndex = OUTPUT, .offset = 0, .length = static_cast<uint32_t>(s)},
194 .dimensions = {},
195 };
196 outputs_info[index] = arg;
197 outputSize += s;
198 });
Xusong Wang929fd212019-01-27 23:08:12 -0800199 // If output0 does not have size larger than one byte,
200 // we can not provide an insufficient buffer
201 if (!sizeLargerThanOne && outputType == OutputType::INSUFFICIENT) return;
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700202 // Compute offset for outputs 1 and so on
203 {
204 size_t offset = 0;
205 for (auto& i : outputs_info) {
206 i.location.offset = offset;
207 offset += i.location.length;
208 }
209 }
Miao Wanga2d04c82018-02-05 17:26:54 -0800210 std::vector<hidl_memory> pools = {nn::allocateSharedMemory(inputSize),
211 nn::allocateSharedMemory(outputSize)};
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700212 ASSERT_NE(0ull, pools[INPUT].size());
213 ASSERT_NE(0ull, pools[OUTPUT].size());
214
215 // load data
216 sp<IMemory> inputMemory = mapMemory(pools[INPUT]);
217 sp<IMemory> outputMemory = mapMemory(pools[OUTPUT]);
218 ASSERT_NE(nullptr, inputMemory.get());
219 ASSERT_NE(nullptr, outputMemory.get());
220 char* inputPtr = reinterpret_cast<char*>(static_cast<void*>(inputMemory->getPointer()));
221 char* outputPtr = reinterpret_cast<char*>(static_cast<void*>(outputMemory->getPointer()));
222 ASSERT_NE(nullptr, inputPtr);
223 ASSERT_NE(nullptr, outputPtr);
224 inputMemory->update();
225 outputMemory->update();
226
227 // Go through all inputs, copy the values
228 for_all(inputs, [&inputs_info, inputPtr](int index, auto p, auto s) {
229 char* begin = (char*)p;
230 char* end = begin + s;
231 // TODO: handle more than one input
232 std::copy(begin, end, inputPtr + inputs_info[index].location.offset);
233 });
234
235 inputMemory->commit();
236 outputMemory->commit();
Michael Butlercf22a572017-09-22 13:26:12 -0700237
Michael Butler814d8372019-01-15 11:02:55 -0800238 const Request request = {.inputs = inputs_info, .outputs = outputs_info, .pools = pools};
239
Xusong Wang187c5972018-11-07 09:33:59 -0800240 ErrorStatus executionStatus;
241 hidl_vec<OutputShape> outputShapes;
David Grosse3013492019-01-23 14:01:52 -0800242 Timing timing;
Michael Butler814d8372019-01-15 11:02:55 -0800243 switch (executor) {
244 case Executor::ASYNC: {
245 SCOPED_TRACE("asynchronous");
Michael Butlercf22a572017-09-22 13:26:12 -0700246
Michael Butler814d8372019-01-15 11:02:55 -0800247 // launch execution
248 sp<ExecutionCallback> executionCallback = new ExecutionCallback();
249 ASSERT_NE(nullptr, executionCallback.get());
250 Return<ErrorStatus> executionLaunchStatus =
251 ExecutePreparedModel(preparedModel, request, measure, executionCallback);
252 ASSERT_TRUE(executionLaunchStatus.isOk());
253 EXPECT_EQ(ErrorStatus::NONE, static_cast<ErrorStatus>(executionLaunchStatus));
David Gross49e41672018-12-21 11:20:26 -0800254
Michael Butler814d8372019-01-15 11:02:55 -0800255 // retrieve execution status
256 executionCallback->wait();
257 executionStatus = executionCallback->getStatus();
258 outputShapes = executionCallback->getOutputShapes();
259 timing = executionCallback->getTiming();
David Gross49e41672018-12-21 11:20:26 -0800260
Michael Butler814d8372019-01-15 11:02:55 -0800261 break;
262 }
263 case Executor::SYNC: {
264 SCOPED_TRACE("synchronous");
265
266 // execute
267 Return<ErrorStatus> executionReturnStatus = ExecutePreparedModel(
268 preparedModel, request, measure, &outputShapes, &timing);
269 ASSERT_TRUE(executionReturnStatus.isOk());
270 executionStatus = static_cast<ErrorStatus>(executionReturnStatus);
271
272 break;
273 }
274 case Executor::BURST: {
275 SCOPED_TRACE("burst");
276
277 // create burst
278 const std::unique_ptr<::android::nn::ExecutionBurstController> controller =
279 CreateBurst(preparedModel);
280 ASSERT_NE(nullptr, controller.get());
281
282 // create memory keys
283 std::vector<intptr_t> keys(request.pools.size());
284 for (size_t i = 0; i < keys.size(); ++i) {
285 keys[i] = reinterpret_cast<intptr_t>(&request.pools[i]);
286 }
287
288 // execute burst
289 std::tie(executionStatus, outputShapes, timing) =
290 controller->compute(request, measure, keys);
291
292 break;
293 }
David Gross49e41672018-12-21 11:20:26 -0800294 }
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700295
Xusong Wang929fd212019-01-27 23:08:12 -0800296 if (outputType != OutputType::FULLY_SPECIFIED &&
297 executionStatus == ErrorStatus::GENERAL_FAILURE) {
Xusong Wanga3165812018-11-19 18:26:08 -0800298 LOG(INFO) << "NN VTS: Early termination of test because vendor service cannot "
299 "execute model that it does not support.";
300 std::cout << "[ ] Early termination of test because vendor service cannot "
301 "execute model that it does not support."
302 << std::endl;
Xusong Wang929fd212019-01-27 23:08:12 -0800303 GTEST_SKIP();
Xusong Wanga3165812018-11-19 18:26:08 -0800304 }
David Grosse3013492019-01-23 14:01:52 -0800305 if (measure == MeasureTiming::NO) {
306 EXPECT_EQ(UINT64_MAX, timing.timeOnDevice);
307 EXPECT_EQ(UINT64_MAX, timing.timeInDriver);
308 } else {
309 if (timing.timeOnDevice != UINT64_MAX && timing.timeInDriver != UINT64_MAX) {
310 EXPECT_LE(timing.timeOnDevice, timing.timeInDriver);
311 }
312 }
Xusong Wanga3165812018-11-19 18:26:08 -0800313
Xusong Wang929fd212019-01-27 23:08:12 -0800314 switch (outputType) {
315 case OutputType::FULLY_SPECIFIED:
316 // If the model output operands are fully specified, outputShapes must be either
317 // either empty, or have the same number of elements as the number of outputs.
318 ASSERT_EQ(ErrorStatus::NONE, executionStatus);
319 ASSERT_TRUE(outputShapes.size() == 0 ||
320 outputShapes.size() == test.operandDimensions.size());
321 break;
322 case OutputType::UNSPECIFIED:
323 // If the model output operands are not fully specified, outputShapes must have
324 // the same number of elements as the number of outputs.
325 ASSERT_EQ(ErrorStatus::NONE, executionStatus);
326 ASSERT_EQ(outputShapes.size(), test.operandDimensions.size());
327 break;
328 case OutputType::INSUFFICIENT:
329 ASSERT_EQ(ErrorStatus::OUTPUT_INSUFFICIENT_SIZE, executionStatus);
330 ASSERT_EQ(outputShapes.size(), test.operandDimensions.size());
331 ASSERT_FALSE(outputShapes[0].isSufficient);
332 return;
333 }
Xusong Wanga3165812018-11-19 18:26:08 -0800334 // Go through all outputs, overwrite output dimensions with returned output shapes
Xusong Wang929fd212019-01-27 23:08:12 -0800335 if (outputShapes.size() > 0) {
Xusong Wanga3165812018-11-19 18:26:08 -0800336 for_each<uint32_t>(test.operandDimensions,
337 [&outputShapes](int idx, std::vector<uint32_t>& dim) {
338 dim = outputShapes[idx].dimensions;
339 });
340 }
Xusong Wang187c5972018-11-07 09:33:59 -0800341
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700342 // validate results
343 outputMemory->read();
I-Jui (Ray) Sungf6b85502017-09-20 13:45:50 -0700344 copy_back(&test, outputs_info, outputPtr);
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700345 outputMemory->commit();
I-Jui (Ray) Sung7d765bd2017-09-13 18:47:12 -0700346 // Filter out don't cares
I-Jui (Ray) Sung5bf4edf2017-10-06 13:22:39 -0700347 MixedTyped filtered_golden = filter(golden, is_ignored);
348 MixedTyped filtered_test = filter(test, is_ignored);
I-Jui (Ray) Sung7d765bd2017-09-13 18:47:12 -0700349
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700350 // We want "close-enough" results for float
Xusong Wang10d77e42018-08-28 16:50:01 -0700351 compare(filtered_golden, filtered_test, fpAtol, fpRtol);
Michael K. Sanders941d61a2018-10-19 14:39:09 +0100352
353 if (example.expectedMultinomialDistributionTolerance > 0) {
354 expectMultinomialDistributionWithinTolerance(test, example);
355 }
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700356 }
357}
David Gross49e41672018-12-21 11:20:26 -0800358template <typename T_IPreparedModel>
359void EvaluatePreparedModel(sp<T_IPreparedModel>& preparedModel, std::function<bool(int)> is_ignored,
360 const std::vector<MixedTypedExample>& examples,
Michael Butler814d8372019-01-15 11:02:55 -0800361 bool hasRelaxedFloat32Model, Executor executor, MeasureTiming measure,
Xusong Wang929fd212019-01-27 23:08:12 -0800362 OutputType outputType) {
David Gross49e41672018-12-21 11:20:26 -0800363 EvaluatePreparedModel(preparedModel, is_ignored, examples, hasRelaxedFloat32Model, kDefaultAtol,
Xusong Wang929fd212019-01-27 23:08:12 -0800364 kDefaultRtol, executor, measure, outputType);
David Gross49e41672018-12-21 11:20:26 -0800365}
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700366
Xusong Wangb5cb8f72018-10-31 08:43:12 -0700367static void getPreparedModel(sp<PreparedModelCallback> callback,
368 sp<V1_0::IPreparedModel>* preparedModel) {
369 *preparedModel = callback->getPreparedModel();
370}
371static void getPreparedModel(sp<PreparedModelCallback> callback,
372 sp<V1_2::IPreparedModel>* preparedModel) {
373 sp<V1_0::IPreparedModel> preparedModelV1_0 = callback->getPreparedModel();
374 *preparedModel = V1_2::IPreparedModel::castFrom(preparedModelV1_0).withDefault(nullptr);
375}
376
Michael Butlerf76acd02018-03-22 16:37:57 -0700377void Execute(const sp<V1_0::IDevice>& device, std::function<V1_0::Model(void)> create_model,
Michael K. Sanders941d61a2018-10-19 14:39:09 +0100378 std::function<bool(int)> is_ignored, const std::vector<MixedTypedExample>& examples) {
Miao Wanga2d04c82018-02-05 17:26:54 -0800379 V1_0::Model model = create_model();
380
381 // see if service can handle model
382 bool fullySupportsModel = false;
Miao Wanga2d04c82018-02-05 17:26:54 -0800383 Return<void> supportedCall = device->getSupportedOperations(
Michael Butler4d5bb102018-02-26 15:24:46 -0800384 model, [&fullySupportsModel](ErrorStatus status, const hidl_vec<bool>& supported) {
385 ASSERT_EQ(ErrorStatus::NONE, status);
Miao Wanga2d04c82018-02-05 17:26:54 -0800386 ASSERT_NE(0ul, supported.size());
387 fullySupportsModel =
388 std::all_of(supported.begin(), supported.end(), [](bool valid) { return valid; });
389 });
390 ASSERT_TRUE(supportedCall.isOk());
Michael Butler4d5bb102018-02-26 15:24:46 -0800391
392 // launch prepare model
393 sp<PreparedModelCallback> preparedModelCallback = new PreparedModelCallback();
394 ASSERT_NE(nullptr, preparedModelCallback.get());
Miao Wanga2d04c82018-02-05 17:26:54 -0800395 Return<ErrorStatus> prepareLaunchStatus = device->prepareModel(model, preparedModelCallback);
396 ASSERT_TRUE(prepareLaunchStatus.isOk());
Michael Butler4d5bb102018-02-26 15:24:46 -0800397 ASSERT_EQ(ErrorStatus::NONE, static_cast<ErrorStatus>(prepareLaunchStatus));
Miao Wanga2d04c82018-02-05 17:26:54 -0800398
399 // retrieve prepared model
400 preparedModelCallback->wait();
401 ErrorStatus prepareReturnStatus = preparedModelCallback->getStatus();
Xusong Wangb5cb8f72018-10-31 08:43:12 -0700402 sp<V1_0::IPreparedModel> preparedModel;
403 getPreparedModel(preparedModelCallback, &preparedModel);
Miao Wanga2d04c82018-02-05 17:26:54 -0800404
405 // early termination if vendor service cannot fully prepare model
Michael Butler4d5bb102018-02-26 15:24:46 -0800406 if (!fullySupportsModel && prepareReturnStatus != ErrorStatus::NONE) {
Miao Wanga2d04c82018-02-05 17:26:54 -0800407 ASSERT_EQ(nullptr, preparedModel.get());
408 LOG(INFO) << "NN VTS: Early termination of test because vendor service cannot "
409 "prepare model that it does not support.";
410 std::cout << "[ ] Early termination of test because vendor service cannot "
411 "prepare model that it does not support."
412 << std::endl;
Miao Wangbb685a42019-01-08 12:27:35 -0800413 GTEST_SKIP();
Miao Wanga2d04c82018-02-05 17:26:54 -0800414 }
Michael Butler4d5bb102018-02-26 15:24:46 -0800415 EXPECT_EQ(ErrorStatus::NONE, prepareReturnStatus);
Miao Wanga2d04c82018-02-05 17:26:54 -0800416 ASSERT_NE(nullptr, preparedModel.get());
417
Xusong Wang10d77e42018-08-28 16:50:01 -0700418 float fpAtol = 1e-5f, fpRtol = 5.0f * 1.1920928955078125e-7f;
Michael K. Sandersefa4c812018-10-30 14:44:48 +0000419 EvaluatePreparedModel(preparedModel, is_ignored, examples,
Michael Butler814d8372019-01-15 11:02:55 -0800420 /*hasRelaxedFloat32Model=*/false, fpAtol, fpRtol, Executor::ASYNC,
Xusong Wang929fd212019-01-27 23:08:12 -0800421 MeasureTiming::NO, OutputType::FULLY_SPECIFIED);
Miao Wanga2d04c82018-02-05 17:26:54 -0800422}
423
Michael Butlerf76acd02018-03-22 16:37:57 -0700424void Execute(const sp<V1_1::IDevice>& device, std::function<V1_1::Model(void)> create_model,
Michael K. Sanders941d61a2018-10-19 14:39:09 +0100425 std::function<bool(int)> is_ignored, const std::vector<MixedTypedExample>& examples) {
Miao Wanga2d04c82018-02-05 17:26:54 -0800426 V1_1::Model model = create_model();
427
428 // see if service can handle model
429 bool fullySupportsModel = false;
Miao Wanga2d04c82018-02-05 17:26:54 -0800430 Return<void> supportedCall = device->getSupportedOperations_1_1(
Michael Butler4d5bb102018-02-26 15:24:46 -0800431 model, [&fullySupportsModel](ErrorStatus status, const hidl_vec<bool>& supported) {
432 ASSERT_EQ(ErrorStatus::NONE, status);
Miao Wanga2d04c82018-02-05 17:26:54 -0800433 ASSERT_NE(0ul, supported.size());
434 fullySupportsModel =
435 std::all_of(supported.begin(), supported.end(), [](bool valid) { return valid; });
436 });
437 ASSERT_TRUE(supportedCall.isOk());
Michael Butler4d5bb102018-02-26 15:24:46 -0800438
439 // launch prepare model
440 sp<PreparedModelCallback> preparedModelCallback = new PreparedModelCallback();
441 ASSERT_NE(nullptr, preparedModelCallback.get());
Michael Butler2504c2f2018-04-11 16:30:09 -0700442 Return<ErrorStatus> prepareLaunchStatus = device->prepareModel_1_1(
443 model, ExecutionPreference::FAST_SINGLE_ANSWER, preparedModelCallback);
Miao Wanga2d04c82018-02-05 17:26:54 -0800444 ASSERT_TRUE(prepareLaunchStatus.isOk());
Michael Butler4d5bb102018-02-26 15:24:46 -0800445 ASSERT_EQ(ErrorStatus::NONE, static_cast<ErrorStatus>(prepareLaunchStatus));
Miao Wanga2d04c82018-02-05 17:26:54 -0800446
447 // retrieve prepared model
448 preparedModelCallback->wait();
449 ErrorStatus prepareReturnStatus = preparedModelCallback->getStatus();
Xusong Wangb5cb8f72018-10-31 08:43:12 -0700450 sp<V1_0::IPreparedModel> preparedModel;
451 getPreparedModel(preparedModelCallback, &preparedModel);
Miao Wanga2d04c82018-02-05 17:26:54 -0800452
453 // early termination if vendor service cannot fully prepare model
Michael Butler4d5bb102018-02-26 15:24:46 -0800454 if (!fullySupportsModel && prepareReturnStatus != ErrorStatus::NONE) {
Miao Wanga2d04c82018-02-05 17:26:54 -0800455 ASSERT_EQ(nullptr, preparedModel.get());
456 LOG(INFO) << "NN VTS: Early termination of test because vendor service cannot "
457 "prepare model that it does not support.";
458 std::cout << "[ ] Early termination of test because vendor service cannot "
459 "prepare model that it does not support."
460 << std::endl;
Miao Wangbb685a42019-01-08 12:27:35 -0800461 GTEST_SKIP();
Miao Wanga2d04c82018-02-05 17:26:54 -0800462 }
Michael Butler4d5bb102018-02-26 15:24:46 -0800463 EXPECT_EQ(ErrorStatus::NONE, prepareReturnStatus);
Miao Wanga2d04c82018-02-05 17:26:54 -0800464 ASSERT_NE(nullptr, preparedModel.get());
465
Michael K. Sandersefa4c812018-10-30 14:44:48 +0000466 EvaluatePreparedModel(preparedModel, is_ignored, examples,
Michael Butler814d8372019-01-15 11:02:55 -0800467 model.relaxComputationFloat32toFloat16, 1e-5f, 1e-5f, Executor::ASYNC,
Xusong Wang929fd212019-01-27 23:08:12 -0800468 MeasureTiming::NO, OutputType::FULLY_SPECIFIED);
Miao Wanga2d04c82018-02-05 17:26:54 -0800469}
470
Slava Shklyaev871be942018-09-12 14:52:02 +0100471// TODO: Reduce code duplication.
472void Execute(const sp<V1_2::IDevice>& device, std::function<V1_2::Model(void)> create_model,
Xusong Wanga3165812018-11-19 18:26:08 -0800473 std::function<bool(int)> is_ignored, const std::vector<MixedTypedExample>& examples,
474 bool testDynamicOutputShape) {
Slava Shklyaev871be942018-09-12 14:52:02 +0100475 V1_2::Model model = create_model();
476
477 // see if service can handle model
478 bool fullySupportsModel = false;
479 Return<void> supportedCall = device->getSupportedOperations_1_2(
480 model, [&fullySupportsModel](ErrorStatus status, const hidl_vec<bool>& supported) {
481 ASSERT_EQ(ErrorStatus::NONE, status);
482 ASSERT_NE(0ul, supported.size());
483 fullySupportsModel =
484 std::all_of(supported.begin(), supported.end(), [](bool valid) { return valid; });
485 });
486 ASSERT_TRUE(supportedCall.isOk());
487
488 // launch prepare model
489 sp<PreparedModelCallback> preparedModelCallback = new PreparedModelCallback();
490 ASSERT_NE(nullptr, preparedModelCallback.get());
491 Return<ErrorStatus> prepareLaunchStatus = device->prepareModel_1_2(
492 model, ExecutionPreference::FAST_SINGLE_ANSWER, preparedModelCallback);
493 ASSERT_TRUE(prepareLaunchStatus.isOk());
494 ASSERT_EQ(ErrorStatus::NONE, static_cast<ErrorStatus>(prepareLaunchStatus));
495
496 // retrieve prepared model
497 preparedModelCallback->wait();
498 ErrorStatus prepareReturnStatus = preparedModelCallback->getStatus();
Xusong Wangb5cb8f72018-10-31 08:43:12 -0700499 sp<V1_2::IPreparedModel> preparedModel;
500 getPreparedModel(preparedModelCallback, &preparedModel);
Slava Shklyaev871be942018-09-12 14:52:02 +0100501
502 // early termination if vendor service cannot fully prepare model
503 if (!fullySupportsModel && prepareReturnStatus != ErrorStatus::NONE) {
504 ASSERT_EQ(nullptr, preparedModel.get());
505 LOG(INFO) << "NN VTS: Early termination of test because vendor service cannot "
506 "prepare model that it does not support.";
507 std::cout << "[ ] Early termination of test because vendor service cannot "
508 "prepare model that it does not support."
509 << std::endl;
Miao Wangbb685a42019-01-08 12:27:35 -0800510 GTEST_SKIP();
Slava Shklyaev871be942018-09-12 14:52:02 +0100511 }
512 EXPECT_EQ(ErrorStatus::NONE, prepareReturnStatus);
513 ASSERT_NE(nullptr, preparedModel.get());
514
Xusong Wang929fd212019-01-27 23:08:12 -0800515 if (testDynamicOutputShape) {
516 EvaluatePreparedModel(preparedModel, is_ignored, examples,
517 model.relaxComputationFloat32toFloat16, Executor::ASYNC,
518 MeasureTiming::NO, OutputType::UNSPECIFIED);
519 EvaluatePreparedModel(preparedModel, is_ignored, examples,
520 model.relaxComputationFloat32toFloat16, Executor::SYNC,
521 MeasureTiming::NO, OutputType::UNSPECIFIED);
522 EvaluatePreparedModel(preparedModel, is_ignored, examples,
523 model.relaxComputationFloat32toFloat16, Executor::BURST,
524 MeasureTiming::NO, OutputType::UNSPECIFIED);
525 EvaluatePreparedModel(preparedModel, is_ignored, examples,
526 model.relaxComputationFloat32toFloat16, Executor::ASYNC,
527 MeasureTiming::YES, OutputType::UNSPECIFIED);
528 EvaluatePreparedModel(preparedModel, is_ignored, examples,
529 model.relaxComputationFloat32toFloat16, Executor::SYNC,
530 MeasureTiming::YES, OutputType::UNSPECIFIED);
531 EvaluatePreparedModel(preparedModel, is_ignored, examples,
532 model.relaxComputationFloat32toFloat16, Executor::BURST,
533 MeasureTiming::YES, OutputType::UNSPECIFIED);
534 EvaluatePreparedModel(preparedModel, is_ignored, examples,
535 model.relaxComputationFloat32toFloat16, Executor::ASYNC,
536 MeasureTiming::NO, OutputType::INSUFFICIENT);
537 EvaluatePreparedModel(preparedModel, is_ignored, examples,
538 model.relaxComputationFloat32toFloat16, Executor::SYNC,
539 MeasureTiming::NO, OutputType::INSUFFICIENT);
540 EvaluatePreparedModel(preparedModel, is_ignored, examples,
541 model.relaxComputationFloat32toFloat16, Executor::BURST,
542 MeasureTiming::NO, OutputType::INSUFFICIENT);
543 EvaluatePreparedModel(preparedModel, is_ignored, examples,
544 model.relaxComputationFloat32toFloat16, Executor::ASYNC,
545 MeasureTiming::YES, OutputType::INSUFFICIENT);
546 EvaluatePreparedModel(preparedModel, is_ignored, examples,
547 model.relaxComputationFloat32toFloat16, Executor::SYNC,
548 MeasureTiming::YES, OutputType::INSUFFICIENT);
549 EvaluatePreparedModel(preparedModel, is_ignored, examples,
550 model.relaxComputationFloat32toFloat16, Executor::BURST,
551 MeasureTiming::YES, OutputType::INSUFFICIENT);
552 } else {
553 EvaluatePreparedModel(preparedModel, is_ignored, examples,
554 model.relaxComputationFloat32toFloat16, Executor::ASYNC,
555 MeasureTiming::NO, OutputType::FULLY_SPECIFIED);
556 EvaluatePreparedModel(preparedModel, is_ignored, examples,
557 model.relaxComputationFloat32toFloat16, Executor::SYNC,
558 MeasureTiming::NO, OutputType::FULLY_SPECIFIED);
559 EvaluatePreparedModel(preparedModel, is_ignored, examples,
560 model.relaxComputationFloat32toFloat16, Executor::BURST,
561 MeasureTiming::NO, OutputType::FULLY_SPECIFIED);
562 EvaluatePreparedModel(preparedModel, is_ignored, examples,
563 model.relaxComputationFloat32toFloat16, Executor::ASYNC,
564 MeasureTiming::YES, OutputType::FULLY_SPECIFIED);
565 EvaluatePreparedModel(preparedModel, is_ignored, examples,
566 model.relaxComputationFloat32toFloat16, Executor::SYNC,
567 MeasureTiming::YES, OutputType::FULLY_SPECIFIED);
568 EvaluatePreparedModel(preparedModel, is_ignored, examples,
569 model.relaxComputationFloat32toFloat16, Executor::BURST,
570 MeasureTiming::YES, OutputType::FULLY_SPECIFIED);
571 }
Slava Shklyaev871be942018-09-12 14:52:02 +0100572}
573
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700574} // namespace generated_tests
575
I-Jui (Ray) Sung2c4e1362017-09-06 02:15:54 -0700576} // namespace neuralnetworks
577} // namespace hardware
578} // namespace android