ArmNN
NotReleased
ClWorkloadFactoryHelper.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include <
armnn/backends/IBackendInternal.hpp
>
9
#include <
armnn/backends/IMemoryManager.hpp
>
10
#include <
backendsCommon/test/WorkloadFactoryHelper.hpp
>
11
12
#include <
cl/ClBackend.hpp
>
13
#include <
cl/ClWorkloadFactory.hpp
>
14
15
#include <boost/polymorphic_pointer_cast.hpp>
16
17
namespace
18
{
19
20
template
<>
21
struct
WorkloadFactoryHelper<
armnn
::ClWorkloadFactory>
22
{
23
static
armnn::IBackendInternal::IMemoryManagerSharedPtr
GetMemoryManager()
24
{
25
armnn::ClBackend
backend;
26
return
backend.
CreateMemoryManager
();
27
}
28
29
static
armnn::ClWorkloadFactory
GetFactory(
30
const
armnn::IBackendInternal::IMemoryManagerSharedPtr
& memoryManager)
31
{
32
return
armnn::ClWorkloadFactory
(boost::polymorphic_pointer_downcast<armnn::ClMemoryManager>(memoryManager));
33
}
34
};
35
36
using
ClWorkloadFactoryHelper = WorkloadFactoryHelper<armnn::ClWorkloadFactory>;
37
38
}
// anonymous namespace
ClBackend.hpp
armnn::IBackendInternal::IMemoryManagerSharedPtr
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
Definition:
IBackendInternal.hpp:90
armnn
Definition:
BackendHelper.hpp:11
ClWorkloadFactory.hpp
IBackendInternal.hpp
armnn::ClBackend
Definition:
ClBackend.hpp:12
IMemoryManager.hpp
armnn::ClBackend::CreateMemoryManager
IBackendInternal::IMemoryManagerUniquePtr CreateMemoryManager() const override
Definition:
ClBackend.cpp:35
armnn::ClWorkloadFactory
Definition:
ClWorkloadFactory.hpp:17
WorkloadFactoryHelper.hpp
src
backends
cl
test
ClWorkloadFactoryHelper.hpp
Generated on Fri Mar 13 2020 16:06:57 for ArmNN by
1.8.13