blob: e138f56fd4dee7be7cb2948064b74f08fa388c91 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
#include <type_traits>
namespace armnn
{
enum class MemorySource
{
Malloc = 1,
DmaBuf = 2,
DmaBufProtected = 4
};
using MemorySourceFlags = unsigned int;
template<typename T>
struct IsMemorySource
{
static const bool value = false;
};
template<>
struct IsMemorySource<MemorySource>
{
static const bool value = true;
};
template <typename Arg, typename std::enable_if<IsMemorySource<Arg>::value>::type* = nullptr>
MemorySourceFlags Combine(Arg sourceA, Arg sourceB)
{
return static_cast<MemorySourceFlags>(sourceA) | static_cast<MemorySourceFlags>(sourceB);
}
template <typename Arg, typename ... Args, typename std::enable_if<IsMemorySource<Arg>::value>::type* = nullptr>
MemorySourceFlags Combine(Arg source, Args... rest)
{
return static_cast<MemorySourceFlags>(source) | Combine(rest...);
}
inline bool CheckFlag(MemorySourceFlags flags, MemorySource source)
{
return (static_cast<MemorySourceFlags>(source) & flags) != 0;
}
} //namespace armnn
|
