diff options
| author | Nikolaj Jensen <nikolaj.jensen@arm.com> | 2023-06-27 14:13:24 +0100 |
|---|---|---|
| committer | Nikolaj Jensen <nikolaj.jensen@arm.com> | 2023-07-26 10:18:54 +0000 |
| commit | fab6c210b37f1fa6b3e37a2583b18f8e4b5a4f12 (patch) | |
| tree | bef025e8bee719fb6649eedc99150f5c066d1bc3 /compute_kernel_writer/prototype/include/ckw/KernelWriter.h | |
| parent | 23882a9014eb3972bca958206866c8e0d0b829cc (diff) | |
| download | ComputeLibrary-fab6c210b37f1fa6b3e37a2583b18f8e4b5a4f12.tar.gz | |
Design wrapper around CKW for easier writing
Signed-off-by: Nikolaj Jensen <nikolaj.jensen@arm.com>
Change-Id: I114cdedcaf05c6abde046741837eeb73b813aa9d
Signed-off-by: Nikolaj Jensen <nikolaj.jensen@arm.com>
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/c/VisualCompute/ComputeLibrary/+/532180
Tested-by: bsgcomp <bsgcomp@arm.com>
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Comments-Addressed: bsgcomp <bsgcomp@arm.com>
Signed-off-by: Nikolaj Jensen <nikolaj.jensen@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9921
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/prototype/include/ckw/KernelWriter.h')
| -rw-r--r-- | compute_kernel_writer/prototype/include/ckw/KernelWriter.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/compute_kernel_writer/prototype/include/ckw/KernelWriter.h b/compute_kernel_writer/prototype/include/ckw/KernelWriter.h index 146fdac53e..c116e62650 100644 --- a/compute_kernel_writer/prototype/include/ckw/KernelWriter.h +++ b/compute_kernel_writer/prototype/include/ckw/KernelWriter.h @@ -230,16 +230,17 @@ public: */ void op_else(const std::function<void()> &body); - /** Write for-loops: `for(; <var> <cond_op> <cond_value>; <update_op> <update_value>) { body }`. + /** Write for-loops: `for(; <var> <cond_op> <cond_value>; <var> <update_op> <update_value>) { body }`. * * @param[in] var_name The name of the variable used in condition. * @param[in] cond_op The relational binary operator used in condition. * @param[in] cond_value_name The value which the variable is compared against. + * @param[in] update_var_name The name of the variable which is updated. * @param[in] update_op The assignment operator used for updating the update value. * @param[in, out] update_value The value which is updated at every iteration. * @param[in] body The body of the for-loop. */ - void op_for_loop(const TileOperand &var_name, BinaryOp cond_op, const TileOperand &cond_value_name, AssignmentOp update_op, const TileOperand &update_value_name, const std::function<void()> &body); + void op_for_loop(const TileOperand &var_name, BinaryOp cond_op, const TileOperand &cond_value_name, const TileOperand &update_var_name, AssignmentOp update_op, const TileOperand &update_value_name, const std::function<void()> &body); /** Write the return statement: `return;` */ |