blob: fe9310f42365f0a4557e7da6de500cddd85f7448 (
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "RefTensorHandle.hpp"
namespace armnn
{
RefTensorHandle::RefTensorHandle(const TensorInfo &tensorInfo, std::shared_ptr<RefMemoryManager> &memoryManager):
m_TensorInfo(tensorInfo),
m_MemoryManager(memoryManager),
m_Pool(nullptr),
m_UnmanagedMemory(nullptr)
{
}
RefTensorHandle::~RefTensorHandle()
{
if (!m_Pool)
{
// unmanaged
::operator delete(m_UnmanagedMemory);
}
}
void RefTensorHandle::Manage()
{
BOOST_ASSERT_MSG(!m_Pool, "RefTensorHandle::Manage() called twice");
BOOST_ASSERT_MSG(!m_UnmanagedMemory, "RefTensorHandle::Manage() called after Allocate()");
m_Pool = m_MemoryManager->Manage(m_TensorInfo.GetNumBytes());
}
void RefTensorHandle::Allocate()
{
if (!m_UnmanagedMemory)
{
if (!m_Pool)
{
// unmanaged
m_UnmanagedMemory = ::operator new(m_TensorInfo.GetNumBytes());
}
else
{
m_MemoryManager->Allocate(m_Pool);
}
}
else
{
throw InvalidArgumentException("RefTensorHandle::Allocate Trying to allocate a RefTensorHandle"
"that already has allocated memory.");
}
}
const void* RefTensorHandle::Map(bool /*unused*/) const
{
return GetPointer();
}
void* RefTensorHandle::GetPointer() const
{
if (m_UnmanagedMemory)
{
return m_UnmanagedMemory;
}
else
{
BOOST_ASSERT_MSG(m_Pool, "RefTensorHandle::GetPointer called on unmanaged, unallocated tensor handle");
return m_MemoryManager->GetPointer(m_Pool);
}
}
void RefTensorHandle::CopyOutTo(void* dest) const
{
const void *src = GetPointer();
BOOST_ASSERT(src);
memcpy(dest, src, m_TensorInfo.GetNumBytes());
}
void RefTensorHandle::CopyInFrom(const void* src)
{
void *dest = GetPointer();
BOOST_ASSERT(dest);
memcpy(dest, src, m_TensorInfo.GetNumBytes());
}
}
|
