From 2d0c2f5700434a4b0c3345c71a3a45825a0e6766 Mon Sep 17 00:00:00 2001
From: Viet-Hoa Do <viet-hoa.do@arm.com>
Date: Thu, 24 Aug 2023 11:48:19 +0100
Subject: Add CKW flow control writing methods

* Structures: if/else/else if, for, return.
* Add corresponding tests.

Partially resolves: COMPMID-6387
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: I2912ccaf46f836907f21bb53fa82bcc1f48dd224
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10199
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
---
 compute_kernel_writer/src/cl/CLHelpers.cpp | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'compute_kernel_writer/src/cl/CLHelpers.cpp')

diff --git a/compute_kernel_writer/src/cl/CLHelpers.cpp b/compute_kernel_writer/src/cl/CLHelpers.cpp
index e12e5e1b13..ff4408b1a3 100644
--- a/compute_kernel_writer/src/cl/CLHelpers.cpp
+++ b/compute_kernel_writer/src/cl/CLHelpers.cpp
@@ -26,6 +26,7 @@
 
 #include "ckw/Error.h"
 #include "ckw/types/DataType.h"
+#include "ckw/types/Operators.h"
 #include "ckw/types/TensorStorageType.h"
 #include "src/types/DataTypeHelpers.h"
 
@@ -145,6 +146,21 @@ std::string cl_get_variable_storagetype_as_string(TensorStorageType storage)
     return res;
 }
 
+std::string cl_get_assignment_op_as_string(AssignmentOp op)
+{
+    switch(op)
+    {
+        case AssignmentOp::Increment:
+            return "+=";
+
+        case AssignmentOp::Decrement:
+            return "-=";
+
+        default:
+            CKW_THROW_MSG("Unsupported assignment operator!");
+    }
+}
+
 std::tuple<bool, std::string> cl_get_unary_op(UnaryOp op)
 {
     switch(op)
-- 
cgit v1.2.1