diff options
Diffstat (limited to 'src/runtime/CL/CLTensor.cpp')
-rw-r--r-- | src/runtime/CL/CLTensor.cpp | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/src/runtime/CL/CLTensor.cpp b/src/runtime/CL/CLTensor.cpp index 732689e7ec..9bbf926b58 100644 --- a/src/runtime/CL/CLTensor.cpp +++ b/src/runtime/CL/CLTensor.cpp @@ -25,8 +25,8 @@ #include "arm_compute/runtime/CL/CLScheduler.h" -using namespace arm_compute; - +namespace arm_compute +{ CLTensor::CLTensor() : _allocator(this) { @@ -76,3 +76,9 @@ void CLTensor::do_unmap(cl::CommandQueue &q) { _allocator.unmap(q, buffer()); } + +void CLTensor::associate_memory_group(arm_compute::IMemoryGroup *memory_group) +{ + _allocator.set_associated_memory_group(memory_group); +} +} // namespace arm_compute
\ No newline at end of file |