diff options
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/test/OptimizerTests.cpp | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/src/armnn/test/OptimizerTests.cpp b/src/armnn/test/OptimizerTests.cpp index a3949affce..a7277b78b5 100644 --- a/src/armnn/test/OptimizerTests.cpp +++ b/src/armnn/test/OptimizerTests.cpp @@ -187,16 +187,16 @@ public: }; template <typename NamePolicy> -class MockBackend : public IBackendInternal +class CustomAllocatorBackend : public IBackendInternal { public: - MockBackend() : + CustomAllocatorBackend() : m_BackendCapabilities(NamePolicy::GetIdStatic(), {{"NullCapability", false}}), m_CustomAllocator(false) {}; - MockBackend(const BackendCapabilities& capabilities) : + CustomAllocatorBackend(const BackendCapabilities& capabilities) : m_BackendCapabilities(capabilities), m_CustomAllocator(false) {}; - ~MockBackend() = default; + ~CustomAllocatorBackend() = default; static const BackendId& GetIdStatic() { @@ -785,9 +785,10 @@ TEST_CASE("BackendHintTest") auto& backendRegistry = BackendRegistryInstance(); - backendRegistry.Register("MockBackend", []() { return std::make_unique<MockBackend<MockPolicy>>(); }); + backendRegistry.Register("MockBackend", []() { return std::make_unique<CustomAllocatorBackend<MockPolicy>>(); }); - backendRegistry.Register("CustomBackend", []() { return std::make_unique<MockBackend<CustomPolicy>>(); }); + backendRegistry.Register("CustomBackend", + []() { return std::make_unique<CustomAllocatorBackend<CustomPolicy>>(); }); // Define the network auto network = INetwork::Create(); |