diff options
Diffstat (limited to 'tests/CL/CLArrayAccessor.h')
-rw-r--r-- | tests/CL/CLArrayAccessor.h | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/tests/CL/CLArrayAccessor.h b/tests/CL/CLArrayAccessor.h index a38907dd78..a516a9d96a 100644 --- a/tests/CL/CLArrayAccessor.h +++ b/tests/CL/CLArrayAccessor.h @@ -60,7 +60,7 @@ public: _array.unmap(); } - int num_values() const override + size_t num_values() const override { return _array.num_values(); } @@ -75,6 +75,13 @@ public: _array.resize(num); } + T &at(size_t index) const override + { + ARM_COMPUTE_ERROR_ON(_array.buffer() == nullptr); + ARM_COMPUTE_ERROR_ON(index >= num_values()); + return _array.buffer()[index]; + } + private: CLArray<T> &_array; }; |