ClRegistryInitializer.cpp

Go to the documentation of this file.

//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
 
#include "ClBackend.hpp"
 
#include <armnn/BackendRegistry.hpp>
 
namespace
{
 
using namespace armnn;
 
static BackendRegistry::StaticRegistryInitializer g_RegisterHelper
{
    BackendRegistryInstance(),
    ClBackend::GetIdStatic(),
    []()
    {
        // Check if we have a CustomMemoryAllocator associated with the backend
        // and if so register it with the backend.
        auto customAllocators = BackendRegistryInstance().GetAllocators();
        auto allocatorIterator = customAllocators.find(ClBackend::GetIdStatic());
        if (allocatorIterator != customAllocators.end())
        {
            return IBackendInternalUniquePtr(new ClBackend(allocatorIterator->second));
        }
        return IBackendInternalUniquePtr(new ClBackend);
    }
};
 
} // Anonymous namespace