blob: 6c3b7b8a895e8e950f9fa40bbd0ffcb7fc8def1a [file] [log] [blame]
Slava Shklyaevfeb87a92018-09-12 14:52:02 +01001/*
2 * Copyright (C) 2018 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
17#define LOG_TAG "neuralnetworks_hidl_hal_test"
18
19#include "VtsHalNeuralnetworks.h"
20
21#include "Callbacks.h"
22
23namespace android {
24namespace hardware {
25namespace neuralnetworks {
26namespace V1_2 {
27
Slava Shklyaevfeb87a92018-09-12 14:52:02 +010028using V1_0::OperandLifeTime;
Slava Shklyaevfeb87a92018-09-12 14:52:02 +010029using V1_1::ExecutionPreference;
30
31namespace vts {
32namespace functional {
33
Xusong Wang1a06e772018-10-31 08:43:12 -070034using ::android::hardware::neuralnetworks::V1_2::implementation::ExecutionCallback;
35using ::android::hardware::neuralnetworks::V1_2::implementation::PreparedModelCallback;
Slava Shklyaevfeb87a92018-09-12 14:52:02 +010036
37///////////////////////// UTILITY FUNCTIONS /////////////////////////
38
39static void validateGetSupportedOperations(const sp<IDevice>& device, const std::string& message,
40 const Model& model) {
41 SCOPED_TRACE(message + " [getSupportedOperations_1_2]");
42
43 Return<void> ret =
44 device->getSupportedOperations_1_2(model, [&](ErrorStatus status, const hidl_vec<bool>&) {
45 EXPECT_EQ(ErrorStatus::INVALID_ARGUMENT, status);
46 });
47 EXPECT_TRUE(ret.isOk());
48}
49
50static void validatePrepareModel(const sp<IDevice>& device, const std::string& message,
51 const Model& model, ExecutionPreference preference) {
52 SCOPED_TRACE(message + " [prepareModel_1_2]");
53
54 sp<PreparedModelCallback> preparedModelCallback = new PreparedModelCallback();
55 ASSERT_NE(nullptr, preparedModelCallback.get());
56 Return<ErrorStatus> prepareLaunchStatus =
57 device->prepareModel_1_2(model, preference, preparedModelCallback);
58 ASSERT_TRUE(prepareLaunchStatus.isOk());
59 ASSERT_EQ(ErrorStatus::INVALID_ARGUMENT, static_cast<ErrorStatus>(prepareLaunchStatus));
60
61 preparedModelCallback->wait();
62 ErrorStatus prepareReturnStatus = preparedModelCallback->getStatus();
63 ASSERT_EQ(ErrorStatus::INVALID_ARGUMENT, prepareReturnStatus);
Xusong Wang1a06e772018-10-31 08:43:12 -070064 sp<IPreparedModel> preparedModel = getPreparedModel_1_2(preparedModelCallback);
Slava Shklyaevfeb87a92018-09-12 14:52:02 +010065 ASSERT_EQ(nullptr, preparedModel.get());
66}
67
68static bool validExecutionPreference(ExecutionPreference preference) {
69 return preference == ExecutionPreference::LOW_POWER ||
70 preference == ExecutionPreference::FAST_SINGLE_ANSWER ||
71 preference == ExecutionPreference::SUSTAINED_SPEED;
72}
73
74// Primary validation function. This function will take a valid model, apply a
75// mutation to it to invalidate the model, then pass it to interface calls that
76// use the model. Note that the model here is passed by value, and any mutation
77// to the model does not leave this function.
78static void validate(const sp<IDevice>& device, const std::string& message, Model model,
79 const std::function<void(Model*)>& mutation,
80 ExecutionPreference preference = ExecutionPreference::FAST_SINGLE_ANSWER) {
81 mutation(&model);
82 if (validExecutionPreference(preference)) {
83 validateGetSupportedOperations(device, message, model);
84 }
85 validatePrepareModel(device, message, model, preference);
86}
87
88// Delete element from hidl_vec. hidl_vec doesn't support a "remove" operation,
89// so this is efficiently accomplished by moving the element to the end and
90// resizing the hidl_vec to one less.
91template <typename Type>
92static void hidl_vec_removeAt(hidl_vec<Type>* vec, uint32_t index) {
93 if (vec) {
94 std::rotate(vec->begin() + index, vec->begin() + index + 1, vec->end());
95 vec->resize(vec->size() - 1);
96 }
97}
98
99template <typename Type>
100static uint32_t hidl_vec_push_back(hidl_vec<Type>* vec, const Type& value) {
101 // assume vec is valid
102 const uint32_t index = vec->size();
103 vec->resize(index + 1);
104 (*vec)[index] = value;
105 return index;
106}
107
108static uint32_t addOperand(Model* model) {
109 return hidl_vec_push_back(&model->operands,
110 {
111 .type = OperandType::INT32,
112 .dimensions = {},
113 .numberOfConsumers = 0,
114 .scale = 0.0f,
115 .zeroPoint = 0,
116 .lifetime = OperandLifeTime::MODEL_INPUT,
117 .location = {.poolIndex = 0, .offset = 0, .length = 0},
118 });
119}
120
121static uint32_t addOperand(Model* model, OperandLifeTime lifetime) {
122 uint32_t index = addOperand(model);
123 model->operands[index].numberOfConsumers = 1;
124 model->operands[index].lifetime = lifetime;
125 return index;
126}
127
128///////////////////////// VALIDATE MODEL OPERAND TYPE /////////////////////////
129
Michael K. Sanders9233dbe2018-10-30 15:16:54 +0000130static const uint32_t invalidOperandTypes[] = {
131 static_cast<uint32_t>(OperandTypeRange::OPERAND_FUNDAMENTAL_MIN) - 1,
132 static_cast<uint32_t>(OperandTypeRange::OPERAND_FUNDAMENTAL_MAX) + 1,
133 static_cast<uint32_t>(OperandTypeRange::OPERAND_OEM_MIN) - 1,
134 static_cast<uint32_t>(OperandTypeRange::OPERAND_OEM_MAX) + 1,
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100135};
136
137static void mutateOperandTypeTest(const sp<IDevice>& device, const Model& model) {
138 for (size_t operand = 0; operand < model.operands.size(); ++operand) {
Michael K. Sanders9233dbe2018-10-30 15:16:54 +0000139 for (uint32_t invalidOperandType : invalidOperandTypes) {
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100140 const std::string message = "mutateOperandTypeTest: operand " +
141 std::to_string(operand) + " set to value " +
142 std::to_string(invalidOperandType);
143 validate(device, message, model, [operand, invalidOperandType](Model* model) {
144 model->operands[operand].type = static_cast<OperandType>(invalidOperandType);
145 });
146 }
147 }
148}
149
150///////////////////////// VALIDATE OPERAND RANK /////////////////////////
151
152static uint32_t getInvalidRank(OperandType type) {
153 switch (type) {
Xusong Wang56666262018-12-05 14:21:51 -0800154 case OperandType::FLOAT16:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100155 case OperandType::FLOAT32:
156 case OperandType::INT32:
157 case OperandType::UINT32:
Lev Proleev08662c62018-10-01 11:18:31 +0100158 case OperandType::BOOL:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100159 return 1;
Michael K. Sanders5dd84122018-10-12 09:10:15 +0100160 case OperandType::TENSOR_FLOAT16:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100161 case OperandType::TENSOR_FLOAT32:
162 case OperandType::TENSOR_INT32:
163 case OperandType::TENSOR_QUANT8_ASYMM:
Lev Proleev217c4072018-11-13 15:42:36 +0000164 case OperandType::TENSOR_QUANT16_SYMM:
Przemyslaw Szczepaniak4766f8b2018-11-08 15:22:17 +0000165 case OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100166 return 0;
167 default:
168 return 0;
169 }
170}
171
172static void mutateOperandRankTest(const sp<IDevice>& device, const Model& model) {
173 for (size_t operand = 0; operand < model.operands.size(); ++operand) {
174 const uint32_t invalidRank = getInvalidRank(model.operands[operand].type);
175 const std::string message = "mutateOperandRankTest: operand " + std::to_string(operand) +
176 " has rank of " + std::to_string(invalidRank);
177 validate(device, message, model, [operand, invalidRank](Model* model) {
178 model->operands[operand].dimensions = std::vector<uint32_t>(invalidRank, 0);
179 });
180 }
181}
182
183///////////////////////// VALIDATE OPERAND SCALE /////////////////////////
184
185static float getInvalidScale(OperandType type) {
186 switch (type) {
Xusong Wang56666262018-12-05 14:21:51 -0800187 case OperandType::FLOAT16:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100188 case OperandType::FLOAT32:
189 case OperandType::INT32:
190 case OperandType::UINT32:
Lev Proleev08662c62018-10-01 11:18:31 +0100191 case OperandType::BOOL:
Michael K. Sanders5dd84122018-10-12 09:10:15 +0100192 case OperandType::TENSOR_FLOAT16:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100193 case OperandType::TENSOR_FLOAT32:
Przemyslaw Szczepaniak4766f8b2018-11-08 15:22:17 +0000194 case OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100195 return 1.0f;
196 case OperandType::TENSOR_INT32:
197 return -1.0f;
198 case OperandType::TENSOR_QUANT8_ASYMM:
Lev Proleev217c4072018-11-13 15:42:36 +0000199 case OperandType::TENSOR_QUANT16_SYMM:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100200 return 0.0f;
201 default:
202 return 0.0f;
203 }
204}
205
206static void mutateOperandScaleTest(const sp<IDevice>& device, const Model& model) {
207 for (size_t operand = 0; operand < model.operands.size(); ++operand) {
208 const float invalidScale = getInvalidScale(model.operands[operand].type);
209 const std::string message = "mutateOperandScaleTest: operand " + std::to_string(operand) +
210 " has scale of " + std::to_string(invalidScale);
211 validate(device, message, model, [operand, invalidScale](Model* model) {
212 model->operands[operand].scale = invalidScale;
213 });
214 }
215}
216
217///////////////////////// VALIDATE OPERAND ZERO POINT /////////////////////////
218
219static std::vector<int32_t> getInvalidZeroPoints(OperandType type) {
220 switch (type) {
Xusong Wang56666262018-12-05 14:21:51 -0800221 case OperandType::FLOAT16:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100222 case OperandType::FLOAT32:
223 case OperandType::INT32:
224 case OperandType::UINT32:
Lev Proleev08662c62018-10-01 11:18:31 +0100225 case OperandType::BOOL:
Michael K. Sanders5dd84122018-10-12 09:10:15 +0100226 case OperandType::TENSOR_FLOAT16:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100227 case OperandType::TENSOR_FLOAT32:
228 case OperandType::TENSOR_INT32:
Przemyslaw Szczepaniak4766f8b2018-11-08 15:22:17 +0000229 case OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100230 return {1};
231 case OperandType::TENSOR_QUANT8_ASYMM:
232 return {-1, 256};
Lev Proleev217c4072018-11-13 15:42:36 +0000233 case OperandType::TENSOR_QUANT16_SYMM:
234 return {-32769, -1, 1, 32768};
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100235 default:
236 return {};
237 }
238}
239
240static void mutateOperandZeroPointTest(const sp<IDevice>& device, const Model& model) {
241 for (size_t operand = 0; operand < model.operands.size(); ++operand) {
242 const std::vector<int32_t> invalidZeroPoints =
243 getInvalidZeroPoints(model.operands[operand].type);
244 for (int32_t invalidZeroPoint : invalidZeroPoints) {
245 const std::string message = "mutateOperandZeroPointTest: operand " +
246 std::to_string(operand) + " has zero point of " +
247 std::to_string(invalidZeroPoint);
248 validate(device, message, model, [operand, invalidZeroPoint](Model* model) {
249 model->operands[operand].zeroPoint = invalidZeroPoint;
250 });
251 }
252 }
253}
254
255///////////////////////// VALIDATE EXTRA ??? /////////////////////////
256
257// TODO: Operand::lifetime
258// TODO: Operand::location
259
260///////////////////////// VALIDATE OPERATION OPERAND TYPE /////////////////////////
261
262static void mutateOperand(Operand* operand, OperandType type) {
263 Operand newOperand = *operand;
264 newOperand.type = type;
265 switch (type) {
Xusong Wang56666262018-12-05 14:21:51 -0800266 case OperandType::FLOAT16:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100267 case OperandType::FLOAT32:
268 case OperandType::INT32:
269 case OperandType::UINT32:
Lev Proleev08662c62018-10-01 11:18:31 +0100270 case OperandType::BOOL:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100271 newOperand.dimensions = hidl_vec<uint32_t>();
272 newOperand.scale = 0.0f;
273 newOperand.zeroPoint = 0;
274 break;
Michael K. Sanders5dd84122018-10-12 09:10:15 +0100275 case OperandType::TENSOR_FLOAT16:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100276 case OperandType::TENSOR_FLOAT32:
277 newOperand.dimensions =
278 operand->dimensions.size() > 0 ? operand->dimensions : hidl_vec<uint32_t>({1});
279 newOperand.scale = 0.0f;
280 newOperand.zeroPoint = 0;
281 break;
282 case OperandType::TENSOR_INT32:
283 newOperand.dimensions =
284 operand->dimensions.size() > 0 ? operand->dimensions : hidl_vec<uint32_t>({1});
285 newOperand.zeroPoint = 0;
286 break;
287 case OperandType::TENSOR_QUANT8_ASYMM:
Lev Proleev217c4072018-11-13 15:42:36 +0000288 case OperandType::TENSOR_QUANT16_SYMM:
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100289 newOperand.dimensions =
290 operand->dimensions.size() > 0 ? operand->dimensions : hidl_vec<uint32_t>({1});
291 newOperand.scale = operand->scale != 0.0f ? operand->scale : 1.0f;
292 break;
Przemyslaw Szczepaniak4766f8b2018-11-08 15:22:17 +0000293 case OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL: {
294 newOperand.dimensions =
295 operand->dimensions.size() > 0 ? operand->dimensions : hidl_vec<uint32_t>({1});
296 newOperand.scale = 0.0f;
297 newOperand.zeroPoint = 0;
298
299 SymmPerChannelQuantParams channelQuant;
300 channelQuant.channelDim = 0;
301 channelQuant.scales = hidl_vec<float>(
302 operand->dimensions.size() > 0 ? static_cast<size_t>(operand->dimensions[0]) : 0);
303 for (size_t i = 0; i < channelQuant.scales.size(); ++i) {
304 channelQuant.scales[i] = 1.0f;
305 }
306 newOperand.extraParams.channelQuant(std::move(channelQuant));
307 } break;
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100308 case OperandType::OEM:
309 case OperandType::TENSOR_OEM_BYTE:
310 default:
311 break;
312 }
313 *operand = newOperand;
314}
315
Xusong Wang1a2492f2018-10-05 11:49:13 -0700316static bool mutateOperationOperandTypeSkip(size_t operand, OperandType type, const Model& model) {
317 // Do not test OEM types
318 if (type == model.operands[operand].type || type == OperandType::OEM ||
319 type == OperandType::TENSOR_OEM_BYTE) {
320 return true;
321 }
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100322 for (const Operation& operation : model.operations) {
Xusong Wang1a2492f2018-10-05 11:49:13 -0700323 // Skip mutateOperationOperandTypeTest for the following operations.
324 // - LSH_PROJECTION's second argument is allowed to have any type.
Michael K. Sanders2a010122018-11-28 10:35:08 +0000325 // - ARGMIN and ARGMAX's first argument can be any of
326 // TENSOR_(FLOAT16|FLOAT32|INT32|QUANT8_ASYMM).
327 // - CAST's argument can be any of TENSOR_(FLOAT16|FLOAT32|INT32|QUANT8_ASYMM).
Xusong Wang1a2492f2018-10-05 11:49:13 -0700328 switch (operation.type) {
329 case OperationType::LSH_PROJECTION: {
330 if (operand == operation.inputs[1]) {
331 return true;
332 }
333 } break;
334 case OperationType::CAST:
335 case OperationType::ARGMAX:
336 case OperationType::ARGMIN: {
Michael K. Sanders2a010122018-11-28 10:35:08 +0000337 if (type == OperandType::TENSOR_FLOAT16 || type == OperandType::TENSOR_FLOAT32 ||
338 type == OperandType::TENSOR_INT32 || type == OperandType::TENSOR_QUANT8_ASYMM) {
Xusong Wang1a2492f2018-10-05 11:49:13 -0700339 return true;
340 }
341 } break;
342 default:
343 break;
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100344 }
345 }
346 return false;
347}
348
349static void mutateOperationOperandTypeTest(const sp<IDevice>& device, const Model& model) {
350 for (size_t operand = 0; operand < model.operands.size(); ++operand) {
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100351 for (OperandType invalidOperandType : hidl_enum_range<OperandType>{}) {
Xusong Wang1a2492f2018-10-05 11:49:13 -0700352 if (mutateOperationOperandTypeSkip(operand, invalidOperandType, model)) {
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100353 continue;
354 }
355 const std::string message = "mutateOperationOperandTypeTest: operand " +
356 std::to_string(operand) + " set to type " +
357 toString(invalidOperandType);
358 validate(device, message, model, [operand, invalidOperandType](Model* model) {
359 mutateOperand(&model->operands[operand], invalidOperandType);
360 });
361 }
362 }
363}
364
365///////////////////////// VALIDATE MODEL OPERATION TYPE /////////////////////////
366
Michael K. Sanders9233dbe2018-10-30 15:16:54 +0000367static const uint32_t invalidOperationTypes[] = {
368 static_cast<uint32_t>(OperationTypeRange::OPERATION_FUNDAMENTAL_MIN) - 1,
369 static_cast<uint32_t>(OperationTypeRange::OPERATION_FUNDAMENTAL_MAX) + 1,
370 static_cast<uint32_t>(OperationTypeRange::OPERATION_OEM_MIN) - 1,
371 static_cast<uint32_t>(OperationTypeRange::OPERATION_OEM_MAX) + 1,
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100372};
373
374static void mutateOperationTypeTest(const sp<IDevice>& device, const Model& model) {
375 for (size_t operation = 0; operation < model.operations.size(); ++operation) {
Michael K. Sanders9233dbe2018-10-30 15:16:54 +0000376 for (uint32_t invalidOperationType : invalidOperationTypes) {
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100377 const std::string message = "mutateOperationTypeTest: operation " +
378 std::to_string(operation) + " set to value " +
379 std::to_string(invalidOperationType);
380 validate(device, message, model, [operation, invalidOperationType](Model* model) {
381 model->operations[operation].type =
382 static_cast<OperationType>(invalidOperationType);
383 });
384 }
385 }
386}
387
388///////////////////////// VALIDATE MODEL OPERATION INPUT OPERAND INDEX /////////////////////////
389
390static void mutateOperationInputOperandIndexTest(const sp<IDevice>& device, const Model& model) {
391 for (size_t operation = 0; operation < model.operations.size(); ++operation) {
392 const uint32_t invalidOperand = model.operands.size();
393 for (size_t input = 0; input < model.operations[operation].inputs.size(); ++input) {
394 const std::string message = "mutateOperationInputOperandIndexTest: operation " +
395 std::to_string(operation) + " input " +
396 std::to_string(input);
397 validate(device, message, model, [operation, input, invalidOperand](Model* model) {
398 model->operations[operation].inputs[input] = invalidOperand;
399 });
400 }
401 }
402}
403
404///////////////////////// VALIDATE MODEL OPERATION OUTPUT OPERAND INDEX /////////////////////////
405
406static void mutateOperationOutputOperandIndexTest(const sp<IDevice>& device, const Model& model) {
407 for (size_t operation = 0; operation < model.operations.size(); ++operation) {
408 const uint32_t invalidOperand = model.operands.size();
409 for (size_t output = 0; output < model.operations[operation].outputs.size(); ++output) {
410 const std::string message = "mutateOperationOutputOperandIndexTest: operation " +
411 std::to_string(operation) + " output " +
412 std::to_string(output);
413 validate(device, message, model, [operation, output, invalidOperand](Model* model) {
414 model->operations[operation].outputs[output] = invalidOperand;
415 });
416 }
417 }
418}
419
420///////////////////////// REMOVE OPERAND FROM EVERYTHING /////////////////////////
421
422static void removeValueAndDecrementGreaterValues(hidl_vec<uint32_t>* vec, uint32_t value) {
423 if (vec) {
424 // remove elements matching "value"
425 auto last = std::remove(vec->begin(), vec->end(), value);
426 vec->resize(std::distance(vec->begin(), last));
427
428 // decrement elements exceeding "value"
429 std::transform(vec->begin(), vec->end(), vec->begin(),
430 [value](uint32_t v) { return v > value ? v-- : v; });
431 }
432}
433
434static void removeOperand(Model* model, uint32_t index) {
435 hidl_vec_removeAt(&model->operands, index);
436 for (Operation& operation : model->operations) {
437 removeValueAndDecrementGreaterValues(&operation.inputs, index);
438 removeValueAndDecrementGreaterValues(&operation.outputs, index);
439 }
440 removeValueAndDecrementGreaterValues(&model->inputIndexes, index);
441 removeValueAndDecrementGreaterValues(&model->outputIndexes, index);
442}
443
Xusong Wang1a2492f2018-10-05 11:49:13 -0700444static bool removeOperandSkip(size_t operand, const Model& model) {
445 for (const Operation& operation : model.operations) {
446 // Skip removeOperandTest for the following operations.
447 // - SPLIT's outputs are not checked during prepareModel.
448 if (operation.type == OperationType::SPLIT) {
449 for (const size_t outOprand : operation.outputs) {
450 if (operand == outOprand) {
451 return true;
452 }
453 }
454 }
455 }
456 return false;
457}
458
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100459static void removeOperandTest(const sp<IDevice>& device, const Model& model) {
460 for (size_t operand = 0; operand < model.operands.size(); ++operand) {
Xusong Wang1a2492f2018-10-05 11:49:13 -0700461 if (removeOperandSkip(operand, model)) {
462 continue;
463 }
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100464 const std::string message = "removeOperandTest: operand " + std::to_string(operand);
465 validate(device, message, model,
466 [operand](Model* model) { removeOperand(model, operand); });
467 }
468}
469
470///////////////////////// REMOVE OPERATION /////////////////////////
471
472static void removeOperation(Model* model, uint32_t index) {
473 for (uint32_t operand : model->operations[index].inputs) {
474 model->operands[operand].numberOfConsumers--;
475 }
476 hidl_vec_removeAt(&model->operations, index);
477}
478
479static void removeOperationTest(const sp<IDevice>& device, const Model& model) {
480 for (size_t operation = 0; operation < model.operations.size(); ++operation) {
481 const std::string message = "removeOperationTest: operation " + std::to_string(operation);
482 validate(device, message, model,
483 [operation](Model* model) { removeOperation(model, operation); });
484 }
485}
486
487///////////////////////// REMOVE OPERATION INPUT /////////////////////////
488
Xusong Wang1a2492f2018-10-05 11:49:13 -0700489static bool removeOperationInputSkip(const Operation& op, size_t input) {
490 // Skip removeOperationInputTest for the following operations.
491 // - CONCATENATION has at least 2 inputs, with the last element being INT32.
492 // - CONV_2D, DEPTHWISE_CONV_2D, MAX_POOL_2D, AVERAGE_POOL_2D, L2_POOL_2D, RESIZE_BILINEAR,
493 // SPACE_TO_DEPTH, SPACE_TO_DEPTH, SPACE_TO_BATCH_ND, BATCH_TO_SPACE_ND can have an optional
494 // layout parameter.
495 // - L2_NORMALIZATION, LOCAL_RESPONSE_NORMALIZATION, SOFTMAX can have an optional axis
496 // parameter.
497 switch (op.type) {
498 case OperationType::CONCATENATION: {
499 if (op.inputs.size() > 2 && input != op.inputs.size() - 1) {
500 return true;
501 }
502 } break;
503 case OperationType::DEPTHWISE_CONV_2D: {
504 if ((op.inputs.size() == 12 && input == 11) || (op.inputs.size() == 9 && input == 8)) {
505 return true;
506 }
507 } break;
508 case OperationType::CONV_2D:
509 case OperationType::AVERAGE_POOL_2D:
510 case OperationType::MAX_POOL_2D:
511 case OperationType::L2_POOL_2D: {
512 if ((op.inputs.size() == 11 && input == 10) || (op.inputs.size() == 8 && input == 7)) {
513 return true;
514 }
515 } break;
516 case OperationType::RESIZE_BILINEAR: {
517 if (op.inputs.size() == 4 && input == 3) {
518 return true;
519 }
520 } break;
521 case OperationType::SPACE_TO_DEPTH:
522 case OperationType::DEPTH_TO_SPACE:
523 case OperationType::BATCH_TO_SPACE_ND: {
524 if (op.inputs.size() == 3 && input == 2) {
525 return true;
526 }
527 } break;
528 case OperationType::SPACE_TO_BATCH_ND: {
529 if (op.inputs.size() == 4 && input == 3) {
530 return true;
531 }
532 } break;
533 case OperationType::L2_NORMALIZATION: {
534 if (op.inputs.size() == 2 && input == 1) {
535 return true;
536 }
537 } break;
538 case OperationType::LOCAL_RESPONSE_NORMALIZATION: {
539 if (op.inputs.size() == 6 && input == 5) {
540 return true;
541 }
542 } break;
543 case OperationType::SOFTMAX: {
544 if (op.inputs.size() == 3 && input == 2) {
545 return true;
546 }
547 } break;
548 default:
549 break;
550 }
551 return false;
552}
553
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100554static void removeOperationInputTest(const sp<IDevice>& device, const Model& model) {
555 for (size_t operation = 0; operation < model.operations.size(); ++operation) {
556 for (size_t input = 0; input < model.operations[operation].inputs.size(); ++input) {
557 const Operation& op = model.operations[operation];
Xusong Wang1a2492f2018-10-05 11:49:13 -0700558 if (removeOperationInputSkip(op, input)) {
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100559 continue;
560 }
561 const std::string message = "removeOperationInputTest: operation " +
562 std::to_string(operation) + ", input " +
563 std::to_string(input);
564 validate(device, message, model, [operation, input](Model* model) {
565 uint32_t operand = model->operations[operation].inputs[input];
566 model->operands[operand].numberOfConsumers--;
567 hidl_vec_removeAt(&model->operations[operation].inputs, input);
568 });
569 }
570 }
571}
572
573///////////////////////// REMOVE OPERATION OUTPUT /////////////////////////
574
575static void removeOperationOutputTest(const sp<IDevice>& device, const Model& model) {
576 for (size_t operation = 0; operation < model.operations.size(); ++operation) {
577 for (size_t output = 0; output < model.operations[operation].outputs.size(); ++output) {
578 const std::string message = "removeOperationOutputTest: operation " +
579 std::to_string(operation) + ", output " +
580 std::to_string(output);
581 validate(device, message, model, [operation, output](Model* model) {
582 hidl_vec_removeAt(&model->operations[operation].outputs, output);
583 });
584 }
585 }
586}
587
588///////////////////////// MODEL VALIDATION /////////////////////////
589
590// TODO: remove model input
591// TODO: remove model output
592// TODO: add unused operation
593
594///////////////////////// ADD OPERATION INPUT /////////////////////////
595
Xusong Wang1a2492f2018-10-05 11:49:13 -0700596static bool addOperationInputSkip(const Operation& op) {
Xusong Wanga6b1dc72018-10-22 13:49:00 -0700597 // Skip addOperationInputTest for the following operations.
Xusong Wang1a2492f2018-10-05 11:49:13 -0700598 // - L2_NORMALIZATION, LOCAL_RESPONSE_NORMALIZATION, SOFTMAX can have an optional INT32 axis
599 // parameter.
600 if ((op.type == OperationType::L2_NORMALIZATION && op.inputs.size() == 1) ||
601 (op.type == OperationType::LOCAL_RESPONSE_NORMALIZATION && op.inputs.size() == 5) ||
602 (op.type == OperationType::SOFTMAX && op.inputs.size() == 2)) {
Xusong Wanga6b1dc72018-10-22 13:49:00 -0700603 return true;
604 }
605 return false;
606}
607
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100608static void addOperationInputTest(const sp<IDevice>& device, const Model& model) {
609 for (size_t operation = 0; operation < model.operations.size(); ++operation) {
Xusong Wanga6b1dc72018-10-22 13:49:00 -0700610 if (addOperationInputSkip(model.operations[operation])) {
611 continue;
612 }
Slava Shklyaevfeb87a92018-09-12 14:52:02 +0100613 const std::string message = "addOperationInputTest: operation " + std::to_string(operation);
614 validate(device, message, model, [operation](Model* model) {
615 uint32_t index = addOperand(model, OperandLifeTime::MODEL_INPUT);
616 hidl_vec_push_back(&model->operations[operation].inputs, index);
617 hidl_vec_push_back(&model->inputIndexes, index);
618 });
619 }
620}
621
622///////////////////////// ADD OPERATION OUTPUT /////////////////////////
623
624static void addOperationOutputTest(const sp<IDevice>& device, const Model& model) {
625 for (size_t operation = 0; operation < model.operations.size(); ++operation) {
626 const std::string message =
627 "addOperationOutputTest: operation " + std::to_string(operation);
628 validate(device, message, model, [operation](Model* model) {
629 uint32_t index = addOperand(model, OperandLifeTime::MODEL_OUTPUT);
630 hidl_vec_push_back(&model->operations[operation].outputs, index);
631 hidl_vec_push_back(&model->outputIndexes, index);
632 });
633 }
634}
635
636///////////////////////// VALIDATE EXECUTION PREFERENCE /////////////////////////
637
638static const int32_t invalidExecutionPreferences[] = {
639 static_cast<int32_t>(ExecutionPreference::LOW_POWER) - 1, // lower bound
640 static_cast<int32_t>(ExecutionPreference::SUSTAINED_SPEED) + 1, // upper bound
641};
642
643static void mutateExecutionPreferenceTest(const sp<IDevice>& device, const Model& model) {
644 for (int32_t preference : invalidExecutionPreferences) {
645 const std::string message =
646 "mutateExecutionPreferenceTest: preference " + std::to_string(preference);
647 validate(device, message, model, [](Model*) {},
648 static_cast<ExecutionPreference>(preference));
649 }
650}
651
652////////////////////////// ENTRY POINT //////////////////////////////
653
654void ValidationTest::validateModel(const Model& model) {
655 mutateOperandTypeTest(device, model);
656 mutateOperandRankTest(device, model);
657 mutateOperandScaleTest(device, model);
658 mutateOperandZeroPointTest(device, model);
659 mutateOperationOperandTypeTest(device, model);
660 mutateOperationTypeTest(device, model);
661 mutateOperationInputOperandIndexTest(device, model);
662 mutateOperationOutputOperandIndexTest(device, model);
663 removeOperandTest(device, model);
664 removeOperationTest(device, model);
665 removeOperationInputTest(device, model);
666 removeOperationOutputTest(device, model);
667 addOperationInputTest(device, model);
668 addOperationOutputTest(device, model);
669 mutateExecutionPreferenceTest(device, model);
670}
671
672} // namespace functional
673} // namespace vts
674} // namespace V1_2
675} // namespace neuralnetworks
676} // namespace hardware
677} // namespace android