diff options
Diffstat (limited to 'compute_kernel_writer/src/KernelWriter.cpp')
-rw-r--r-- | compute_kernel_writer/src/KernelWriter.cpp | 7 |
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) |