aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/GLES_COMPUTE/GCTensor.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/GLES_COMPUTE/GCTensor.cpp')
-rw-r--r--src/runtime/GLES_COMPUTE/GCTensor.cpp14
1 files changed, 10 insertions, 4 deletions
diff --git a/src/runtime/GLES_COMPUTE/GCTensor.cpp b/src/runtime/GLES_COMPUTE/GCTensor.cpp
index e193d26f0a..66c1abdb6d 100644
--- a/src/runtime/GLES_COMPUTE/GCTensor.cpp
+++ b/src/runtime/GLES_COMPUTE/GCTensor.cpp
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2017-2018 ARM Limited.
+ * Copyright (c) 2017-2019 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -24,8 +24,8 @@
#include "arm_compute/runtime/GLES_COMPUTE/GCTensor.h"
-using namespace arm_compute;
-
+namespace arm_compute
+{
GCTensor::GCTensor()
: _allocator(this)
{
@@ -56,6 +56,11 @@ GLuint GCTensor::gc_buffer() const
return _allocator.get_gl_ssbo_name();
}
+void GCTensor::associate_memory_group(arm_compute::IMemoryGroup *memory_group)
+{
+ _allocator.set_associated_memory_group(memory_group);
+}
+
void GCTensor::map(bool blocking)
{
IGCTensor::map(blocking);
@@ -74,4 +79,5 @@ uint8_t *GCTensor::do_map(bool blocking)
void GCTensor::do_unmap()
{
_allocator.unmap();
-} \ No newline at end of file
+}
+} // namespace arm_compute \ No newline at end of file