aboutsummaryrefslogtreecommitdiff
path: root/compute_kernel_writer/src/KernelWriter.cpp
diff options
context:
space:
mode:
authorViet-Hoa Do <viet-hoa.do@arm.com>2023-09-19 16:41:34 +0100
committerViet-Hoa Do <viet-hoa.do@arm.com>2023-09-22 12:07:09 +0000
commitcd1f03e765ad0f3ca3b68b1a7c1d0a1539cab439 (patch)
tree9cb78579e01e14501c316f5297c804ba13c8ad37 /compute_kernel_writer/src/KernelWriter.cpp
parent1f841a52f9a7f52948d676bc3807461bbed6f70a (diff)
downloadComputeLibrary-cd1f03e765ad0f3ca3b68b1a7c1d0a1539cab439.tar.gz
Add row vector and scalar access support to tile operand
* Add the concept of tile view which refers to a specific rectangular area of the tile object. - The active area is added to TileOperand so that the user can access part of the tile. - Currently only row vector and scalar access are exposed to the user. - All writing operations except load/store op support sub-tile. * Add tests for sub-tile access. Resolves: COMPMID-6557 Signed-off-by: Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com> Change-Id: Ica3f9eaf17f06e080c495d36c572f623b62c2910 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10354 Reviewed-by: Gunes Bayir <gunes.bayir@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'compute_kernel_writer/src/KernelWriter.cpp')
-rw-r--r--compute_kernel_writer/src/KernelWriter.cpp7
1 files changed, 5 insertions, 2 deletions
diff --git a/compute_kernel_writer/src/KernelWriter.cpp b/compute_kernel_writer/src/KernelWriter.cpp
index fb0e62c8ce..0bea1200b7 100644
--- a/compute_kernel_writer/src/KernelWriter.cpp
+++ b/compute_kernel_writer/src/KernelWriter.cpp
@@ -27,10 +27,13 @@
#include "ckw/TileOperand.h"
#include "ckw/types/TargetArchitecture.h"
#include "ckw/types/TargetLanguage.h"
+#include "src/TileView.h"
#include "src/cl/CLKernelWriter.h"
#include "src/cl/CLTensorArgument.h"
#include "src/cl/CLTile.h"
+#include <tuple>
+
namespace ckw
{
@@ -90,9 +93,9 @@ TileOperand KernelWriter::create_tile_operand(ITile &tile)
return TileOperand(tile);
}
-ITile &KernelWriter::get_tile(const TileOperand &operand)
+std::tuple<ITile &, TileArea> KernelWriter::get_tile(const TileOperand &operand)
{
- return operand._tile;
+ return { *operand._tile, { operand._row_start, operand._row_end, operand._col_start, operand._col_end } };
}
TensorOperand KernelWriter::create_tensor_operand(ITensor &tensor)