tequilaOS/platform_hardware_interfaces
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge "Make prepareModel asynchronous." into oc-mr1-dev

Browse source
This commit is contained in:
Michael Butler 2017-09-12 18:41:00 +00:00 committed by Android (Google) Code Review
commit 7621f6f5ed
2 changed files with 13 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
neuralnetworks/1.0/IDevice.hal
View file

@ -18,6 +18,7 @@
package android.hardware.neuralnetworks@1.0;
import IEvent;
import IPreparedModel;
interface IDevice {
@ -25,7 +26,7 @@ interface IDevice {
getSupportedSubgraph(Model model) generates(vec<bool> supported);
prepareModel(Model model) generates(IPreparedModel preparedModel);
prepareModel(Model model, IEvent event) generates(IPreparedModel preparedModel);
getStatus() generates(DeviceStatus status);
};

18
neuralnetworks/1.0/vts/functional/VtsHalNeuralnetworksV1_0TargetTest.cpp
View file

@ -209,10 +209,14 @@ TEST_F(NeuralnetworksHidlTest, SimpleExecuteGraphTest) {
const uint32_t INPUT = 0;
const uint32_t OUTPUT = 1;
// prpeare request
// prepare request
Model model = createTestModel();
sp<IPreparedModel> preparedModel = device->prepareModel(model);
sp<Event> preparationEvent = new Event();
ASSERT_NE(nullptr, preparationEvent.get());
sp<IPreparedModel> preparedModel = device->prepareModel(model, preparationEvent);
ASSERT_NE(nullptr, preparedModel.get());
Event::Status preparationStatus = preparationEvent->wait();
EXPECT_EQ(Event::Status::SUCCESS, preparationStatus);
// prepare inputs
uint32_t inputSize = static_cast<uint32_t>(inputData.size() * sizeof(float));
@ -245,13 +249,13 @@ TEST_F(NeuralnetworksHidlTest, SimpleExecuteGraphTest) {
outputMemory->commit();
// execute request
sp<Event> event = sp<Event>(new Event());
ASSERT_NE(nullptr, event.get());
sp<Event> executionEvent = new Event();
ASSERT_NE(nullptr, executionEvent.get());
bool success = preparedModel->execute({.inputs = inputs, .outputs = outputs, .pools = pools},
event);
executionEvent);
EXPECT_TRUE(success);
Event::Status status = event->wait();
EXPECT_EQ(Event::Status::SUCCESS, status);
Event::Status executionStatus = executionEvent->wait();
EXPECT_EQ(Event::Status::SUCCESS, executionStatus);
// validate results { 1+5, 2+6, 3+7, 4+8 }
outputMemory->read();