diff options
Diffstat (limited to 'nn/runtime/test/TestIntrospectionControl.cpp')
| -rw-r--r-- | nn/runtime/test/TestIntrospectionControl.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/nn/runtime/test/TestIntrospectionControl.cpp b/nn/runtime/test/TestIntrospectionControl.cpp index 84617d9b0..9d0cbe6c3 100644 --- a/nn/runtime/test/TestIntrospectionControl.cpp +++ b/nn/runtime/test/TestIntrospectionControl.cpp @@ -16,6 +16,7 @@ #include <gtest/gtest.h> +#include <chrono> #include <iterator> #include <map> #include <queue> @@ -309,7 +310,8 @@ std::set<Success> expectedPassSet = {Success::PASS_NEITHER, Success::PASS_DEVICE class TestPreparedModel12 : public SamplePreparedModel { public: TestPreparedModel12(const HidlModel& model, const SampleDriver* driver, Success success) - : SamplePreparedModel(model, driver), mSuccess(success) {} + : SamplePreparedModel(model, driver, ExecutionPreference::FAST_SINGLE_ANSWER), + mSuccess(success) {} Return<ErrorStatus> execute(const Request&, const sp<V1_0::IExecutionCallback>& callback) override { @@ -384,8 +386,8 @@ class TestPreparedModel12 : public SamplePreparedModel { const MQDescriptorSync<V1_2::FmqRequestDatum>& requestChannel, const MQDescriptorSync<V1_2::FmqResultDatum>& resultChannel, configureExecutionBurst_cb cb) override { - const sp<V1_2::IBurstContext> burst = - ExecutionBurstServer::create(callback, requestChannel, resultChannel, this); + const sp<V1_2::IBurstContext> burst = ExecutionBurstServer::create( + callback, requestChannel, resultChannel, this, std::chrono::microseconds{0}); cb(burst == nullptr ? ErrorStatus::GENERAL_FAILURE : ErrorStatus::NONE, burst); return Void(); |