Compute Library
 21.02
CLLutAllocator.cpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2016, 2017 Arm Limited.
3  *
4  * SPDX-License-Identifier: MIT
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to
8  * deal in the Software without restriction, including without limitation the
9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10  * sell copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
25 
26 #include "arm_compute/core/Error.h"
28 
29 using namespace arm_compute;
30 
32  : _buffer(), _mapping(nullptr)
33 {
34 }
35 
37 {
38  return _mapping;
39 }
40 
41 const cl::Buffer &CLLutAllocator::cl_data() const
42 {
43  return _buffer;
44 }
45 
46 uint8_t *CLLutAllocator::map(cl::CommandQueue &q, bool blocking)
47 {
48  ARM_COMPUTE_ERROR_ON(_buffer.get() == nullptr);
49  return static_cast<uint8_t *>(q.enqueueMapBuffer(_buffer, blocking ? CL_TRUE : CL_FALSE, CL_MAP_READ | CL_MAP_WRITE, 0, size()));
50 }
51 
52 void CLLutAllocator::unmap(cl::CommandQueue &q, uint8_t *mapping)
53 {
54  ARM_COMPUTE_ERROR_ON(_buffer.get() == nullptr);
55  q.enqueueUnmapMemObject(_buffer, mapping);
56 }
57 
58 void CLLutAllocator::allocate()
59 {
60  _buffer = cl::Buffer(CLScheduler::get().context(), CL_MEM_ALLOC_HOST_PTR | CL_MEM_READ_WRITE, size());
61 }
62 
63 uint8_t *CLLutAllocator::lock()
64 {
65  ARM_COMPUTE_ERROR_ON(_mapping != nullptr);
66  cl::CommandQueue q = CLScheduler::get().queue();
67  _mapping = map(q, true);
68  return _mapping;
69 }
70 
71 void CLLutAllocator::unlock()
72 {
73  ARM_COMPUTE_ERROR_ON(_mapping == nullptr);
74  cl::CommandQueue q = CLScheduler::get().queue();
75  unmap(q, _mapping);
76  _mapping = nullptr;
77 }
static CLScheduler & get()
Access the scheduler singleton.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
Definition: Error.h:466
const cl::Buffer & cl_data() const
Interface to be implemented by the child class to return the pointer to the CL data.
Copyright (c) 2017-2021 Arm Limited.
Interface to enqueue OpenCL kernels and get/set the OpenCL CommandQueue and ICLTuner.
CLLutAllocator()
Default constructor.
cl::CommandQueue & queue()
Accessor for the associated CL command queue.
Definition: CLScheduler.cpp:41
uint8_t * data()
Interface to be implemented by the child class to return the pointer to the mapped data...
uint8_t * map(cl::CommandQueue &q, bool blocking)
Enqueue a map operation of the allocated buffer on the given queue.
size_t size() const
Returns the total size in bytes of the LUT.
void unmap(cl::CommandQueue &q, uint8_t *mapping)
Enqueue an unmap operation of the allocated buffer on the given queue.