aboutsummaryrefslogtreecommitdiff
path: root/delegate/opaque/src/test/ArmnnOpaqueDelegateTest.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/opaque/src/test/ArmnnOpaqueDelegateTest.cpp')
-rw-r--r--delegate/opaque/src/test/ArmnnOpaqueDelegateTest.cpp25
1 files changed, 25 insertions, 0 deletions
diff --git a/delegate/opaque/src/test/ArmnnOpaqueDelegateTest.cpp b/delegate/opaque/src/test/ArmnnOpaqueDelegateTest.cpp
index 1635b65809..79f98a9e5e 100644
--- a/delegate/opaque/src/test/ArmnnOpaqueDelegateTest.cpp
+++ b/delegate/opaque/src/test/ArmnnOpaqueDelegateTest.cpp
@@ -9,6 +9,9 @@
#include <opaque/include/armnn_delegate.hpp>
#include <opaque/include/Version.hpp>
+#include <flatbuffers/flatbuffers.h>
+#include <tensorflow/lite/experimental/acceleration/configuration/delegate_registry.h>
+
namespace armnnOpaqueDelegate
{
@@ -37,6 +40,28 @@ TEST_CASE ("DelegateOptions_OpaqueDelegateDefault")
armnnOpaqueDelegate::TfLiteArmnnOpaqueDelegateDelete(opaqueDelegate);
}
+TEST_CASE ("DelegatePluginTest")
+{
+ // Use default settings until options have been enabled.
+ flatbuffers::FlatBufferBuilder flatBufferBuilder;
+ tflite::TFLiteSettingsBuilder tfliteSettingBuilder(flatBufferBuilder);
+ flatbuffers::Offset<tflite::TFLiteSettings> tfliteSettings = tfliteSettingBuilder.Finish();
+ flatBufferBuilder.Finish(tfliteSettings);
+ const tflite::TFLiteSettings* settings = flatbuffers::GetRoot<tflite::TFLiteSettings>(
+ flatBufferBuilder.GetBufferPointer());
+
+ std::unique_ptr<tflite::delegates::DelegatePluginInterface> delegatePlugin =
+ tflite::delegates::DelegatePluginRegistry::CreateByName("armnn_delegate", *settings);
+
+ // Plugin is created correctly using armnn_delegate name.
+ CHECK((delegatePlugin != nullptr));
+
+ tflite::delegates::TfLiteDelegatePtr armnnDelegate = delegatePlugin->Create();
+
+ // Armnn Opaque Delegate is created correctly.
+ CHECK((armnnDelegate != nullptr));
+ CHECK((armnnDelegate->opaque_delegate_builder != nullptr));
}
+}
} // namespace armnnDelegate