diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-05-08 10:18:05 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2019-05-08 10:18:05 +0100 |
commit | e6e54a8b8eae7200fb5501052b13f740fd0b51b9 (patch) | |
tree | 33fdea7a37a9f754b73984cd2b0afe6e1e38f38d /test/DriverTestHelpers.cpp | |
parent | 5e90935244f6448973f81146b78e20a065cb2df6 (diff) | |
download | android-nn-driver-e6e54a8b8eae7200fb5501052b13f740fd0b51b9.tar.gz |
IVGCVSW-2911 Android Q compatibility code updates
* Updates on test classes.
Change-Id: I52e11a1d97d02c23e0a33e4e128dc43a4a95b5bc
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Diffstat (limited to 'test/DriverTestHelpers.cpp')
-rw-r--r-- | test/DriverTestHelpers.cpp | 22 |
1 files changed, 11 insertions, 11 deletions
diff --git a/test/DriverTestHelpers.cpp b/test/DriverTestHelpers.cpp index 3bc0a20a..7a6b7804 100644 --- a/test/DriverTestHelpers.cpp +++ b/test/DriverTestHelpers.cpp @@ -54,7 +54,7 @@ Return<void> ExecutionCallback::wait() } Return<void> PreparedModelCallback::notify(ErrorStatus status, - const android::sp<IPreparedModel>& preparedModel) + const android::sp<V1_0::IPreparedModel>& preparedModel) { m_ErrorStatus = status; m_PreparedModel = preparedModel; @@ -109,10 +109,10 @@ void AddPoolAndSetData(uint32_t size, Request& request, const float* data) memcpy(dst, data, size * sizeof(float)); } -android::sp<IPreparedModel> PrepareModelWithStatus(const V1_0::Model& model, - armnn_driver::ArmnnDriver& driver, - ErrorStatus& prepareStatus, - ErrorStatus expectedStatus) +android::sp<V1_0::IPreparedModel> PrepareModelWithStatus(const V1_0::Model& model, + armnn_driver::ArmnnDriver& driver, + ErrorStatus& prepareStatus, + ErrorStatus expectedStatus) { android::sp<PreparedModelCallback> cb(new PreparedModelCallback()); driver.prepareModel(model, cb); @@ -128,10 +128,10 @@ android::sp<IPreparedModel> PrepareModelWithStatus(const V1_0::Model& model, #ifdef ARMNN_ANDROID_NN_V1_1 -android::sp<IPreparedModel> PrepareModelWithStatus(const V1_1::Model& model, - armnn_driver::ArmnnDriver& driver, - ErrorStatus& prepareStatus, - ErrorStatus expectedStatus) +android::sp<V1_0::IPreparedModel> PrepareModelWithStatus(const V1_1::Model& model, + armnn_driver::ArmnnDriver& driver, + ErrorStatus& prepareStatus, + ErrorStatus expectedStatus) { android::sp<PreparedModelCallback> cb(new PreparedModelCallback()); driver.prepareModel_1_1(model, V1_1::ExecutionPreference::LOW_POWER, cb); @@ -147,7 +147,7 @@ android::sp<IPreparedModel> PrepareModelWithStatus(const V1_1::Model& model, #endif -ErrorStatus Execute(android::sp<IPreparedModel> preparedModel, +ErrorStatus Execute(android::sp<V1_0::IPreparedModel> preparedModel, const Request& request, ErrorStatus expectedStatus) { @@ -160,7 +160,7 @@ ErrorStatus Execute(android::sp<IPreparedModel> preparedModel, return execStatus; } -android::sp<ExecutionCallback> ExecuteNoWait(android::sp<IPreparedModel> preparedModel, const Request& request) +android::sp<ExecutionCallback> ExecuteNoWait(android::sp<V1_0::IPreparedModel> preparedModel, const Request& request) { android::sp<ExecutionCallback> cb(new ExecutionCallback()); BOOST_TEST(preparedModel->execute(request, cb) == ErrorStatus::NONE); |