From 6940dd720ebb6b3d1df8ca203ab696daefe58189 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 20 Mar 2020 12:25:56 +0000 Subject: renamed Documentation folder 20.02 and added .nojekyll file Signed-off-by: Jim Flynn --- 20.02/_buffer_manager_8cpp_source.xhtml | 128 ++++++++++++++++++++++++++++++++ 1 file changed, 128 insertions(+) create mode 100644 20.02/_buffer_manager_8cpp_source.xhtml (limited to '20.02/_buffer_manager_8cpp_source.xhtml') diff --git a/20.02/_buffer_manager_8cpp_source.xhtml b/20.02/_buffer_manager_8cpp_source.xhtml new file mode 100644 index 0000000000..14d4dedacf --- /dev/null +++ b/20.02/_buffer_manager_8cpp_source.xhtml @@ -0,0 +1,128 @@ + + + + + + + + + + + + + +ArmNN: src/profiling/BufferManager.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
BufferManager.cpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2019 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "BufferManager.hpp"
7 #include "PacketBuffer.hpp"
8 
9 namespace armnn
10 {
11 
12 namespace profiling
13 {
14 
15 BufferManager::BufferManager(unsigned int numberOfBuffers, unsigned int maxPacketSize)
16  : m_MaxBufferSize(maxPacketSize),
17  m_NumberOfBuffers(numberOfBuffers)
18 {
19  Initialize();
20 }
21 
22 IPacketBufferPtr BufferManager::Reserve(unsigned int requestedSize, unsigned int& reservedSize)
23 {
24  reservedSize = 0;
25  std::unique_lock<std::mutex> availableListLock(m_AvailableMutex, std::defer_lock);
26  if (requestedSize > m_MaxBufferSize)
27  {
28  return nullptr;
29  }
30  availableListLock.lock();
31  if (m_AvailableList.empty())
32  {
33  availableListLock.unlock();
34  return nullptr;
35  }
36  IPacketBufferPtr buffer = std::move(m_AvailableList.back());
37  m_AvailableList.pop_back();
38  availableListLock.unlock();
39  reservedSize = requestedSize;
40  return buffer;
41 }
42 
43 void BufferManager::Commit(IPacketBufferPtr& packetBuffer, unsigned int size, bool notifyConsumer)
44 {
45  std::unique_lock<std::mutex> readableListLock(m_ReadableMutex, std::defer_lock);
46  packetBuffer->Commit(size);
47  readableListLock.lock();
48  m_ReadableList.push_back(std::move(packetBuffer));
49  readableListLock.unlock();
50 
51  if (notifyConsumer)
52  {
53  FlushReadList();
54  }
55 }
56 
57 void BufferManager::Initialize()
58 {
59  m_AvailableList.reserve(m_NumberOfBuffers);
60  for (unsigned int i = 0; i < m_NumberOfBuffers; ++i)
61  {
62  IPacketBufferPtr buffer = std::make_unique<PacketBuffer>(m_MaxBufferSize);
63  m_AvailableList.emplace_back(std::move(buffer));
64  }
65  m_ReadableList.reserve(m_NumberOfBuffers);
66 }
67 
68 void BufferManager::Release(IPacketBufferPtr& packetBuffer)
69 {
70  std::unique_lock<std::mutex> availableListLock(m_AvailableMutex, std::defer_lock);
71  packetBuffer->Release();
72  availableListLock.lock();
73  m_AvailableList.push_back(std::move(packetBuffer));
74  availableListLock.unlock();
75 }
76 
77 void BufferManager::Reset()
78 {
79  //This method should only be called once all threads have been joined
80  std::lock_guard<std::mutex> readableListLock(m_ReadableMutex);
81  std::lock_guard<std::mutex> availableListLock(m_AvailableMutex);
82 
83  m_AvailableList.clear();
84  m_ReadableList.clear();
85 
86  Initialize();
87 }
88 
89 IPacketBufferPtr BufferManager::GetReadableBuffer()
90 {
91  std::unique_lock<std::mutex> readableListLock(m_ReadableMutex);
92  if (!m_ReadableList.empty())
93  {
94  IPacketBufferPtr buffer = std::move(m_ReadableList.back());
95  m_ReadableList.pop_back();
96  readableListLock.unlock();
97  return buffer;
98  }
99  return nullptr;
100 }
101 
102 void BufferManager::MarkRead(IPacketBufferPtr& packetBuffer)
103 {
104  std::unique_lock<std::mutex> availableListLock(m_AvailableMutex, std::defer_lock);
105  packetBuffer->MarkRead();
106  availableListLock.lock();
107  m_AvailableList.push_back(std::move(packetBuffer));
108  availableListLock.unlock();
109 }
110 
111 void BufferManager::SetConsumer(IConsumer* consumer)
112 {
113  m_Consumer = consumer;
114 }
115 
116 void BufferManager::FlushReadList()
117 {
118  // notify consumer that packet is ready to read
119  if (m_Consumer != nullptr)
120  {
121  m_Consumer->SetReadyToRead();
122  }
123 }
124 
125 } // namespace profiling
126 
127 } // namespace armnn
BufferManager.hpp
+
armnn::profiling::IConsumer::SetReadyToRead
virtual void SetReadyToRead()=0
Set a "ready to read" flag in the buffer to notify the reading thread to start reading it...
+
PacketBuffer.hpp
+ +
armnn::profiling::BufferManager::SetConsumer
void SetConsumer(IConsumer *consumer) override
Set Consumer on the buffer manager to be notified when there is a Commit Can only be one consumer...
Definition: BufferManager.cpp:111
+
armnn
Copyright (c) 2020 ARM Limited.
Definition: 00_introduction.dox:25
+
armnn::profiling::BufferManager::BufferManager
BufferManager(unsigned int numberOfBuffers=5, unsigned int maxPacketSize=4096)
Definition: BufferManager.cpp:15
+
armnn::profiling::BufferManager::FlushReadList
void FlushReadList() override
Notify the Consumer buffer can be read.
Definition: BufferManager.cpp:116
+
armnn::profiling::BufferManager::Commit
void Commit(IPacketBufferPtr &packetBuffer, unsigned int size, bool notifyConsumer=true) override
Definition: BufferManager.cpp:43
+
armnn::profiling::BufferManager::Reserve
IPacketBufferPtr Reserve(unsigned int requestedSize, unsigned int &reservedSize) override
Definition: BufferManager.cpp:22
+
armnn::profiling::BufferManager::GetReadableBuffer
IPacketBufferPtr GetReadableBuffer() override
Definition: BufferManager.cpp:89
+
armnn::profiling::BufferManager::MarkRead
void MarkRead(IPacketBufferPtr &packetBuffer) override
Definition: BufferManager.cpp:102
+ +
armnn::profiling::IPacketBufferPtr
std::unique_ptr< IPacketBuffer > IPacketBufferPtr
Definition: IPacketBuffer.hpp:40
+
armnn::profiling::BufferManager::Release
void Release(IPacketBufferPtr &packetBuffer) override
Definition: BufferManager.cpp:68
+
+
+ + + + -- cgit v1.2.1