aboutsummaryrefslogtreecommitdiff
path: root/compute_kernel_writer/src/cl/CLHelpers.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'compute_kernel_writer/src/cl/CLHelpers.cpp')
-rw-r--r--compute_kernel_writer/src/cl/CLHelpers.cpp16
1 files changed, 16 insertions, 0 deletions
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)