From 366514d54fd43321f440ebddc49fda28d40b8e49 Mon Sep 17 00:00:00 2001 From: Gunes Bayir Date: Thu, 27 Jul 2023 22:52:32 +0100 Subject: Add op_write_raw_code to KernelWriter classes op_write_raw_code helps writing plain text to kernel body. It's used to address the cases where the user needs to explicitly add a code where it's not (yet) supported by the kernel writer utility calls. Partially Resolves: COMPMID-5791 Signed-off-by: Gunes Bayir Change-Id: I04288457ac1871aea8bf10276bf9f078e17fc4b4 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9996 Tested-by: Arm Jenkins Reviewed-by: Gian Marco Iodice Comments-Addressed: Arm Jenkins --- compute_kernel_writer/include/ckw/KernelWriter.h | 9 +++++++++ compute_kernel_writer/src/cl/CLKernelWriter.cpp | 5 +++++ compute_kernel_writer/src/cl/CLKernelWriter.h | 4 ++++ 3 files changed, 18 insertions(+) diff --git a/compute_kernel_writer/include/ckw/KernelWriter.h b/compute_kernel_writer/include/ckw/KernelWriter.h index 894f7b6758..2a347e9ae0 100644 --- a/compute_kernel_writer/include/ckw/KernelWriter.h +++ b/compute_kernel_writer/include/ckw/KernelWriter.h @@ -119,6 +119,15 @@ public: */ virtual TileOperand declare_tile(const std::string &name, const TileInfo &tile_info) = 0; + /** Write the given raw code to kernel source code + * It's used to address the cases where the user needs to + * explicitly add a code where it's not (yet) supported by + * the kernel writer utility calls. + * + * @param[in] raw_code raw code to write as string + */ + virtual void op_write_raw_code(const std::string &raw_code) = 0; + protected: int32_t id_space() const; diff --git a/compute_kernel_writer/src/cl/CLKernelWriter.cpp b/compute_kernel_writer/src/cl/CLKernelWriter.cpp index 69b5244aa2..9363076901 100644 --- a/compute_kernel_writer/src/cl/CLKernelWriter.cpp +++ b/compute_kernel_writer/src/cl/CLKernelWriter.cpp @@ -98,4 +98,9 @@ TileOperand CLKernelWriter::declare_tile(const std::string &name, const TileInfo return operand; } +void CLKernelWriter::op_write_raw_code(const std::string &raw_code) +{ + append_code(raw_code); +} + } // namespace ckw diff --git a/compute_kernel_writer/src/cl/CLKernelWriter.h b/compute_kernel_writer/src/cl/CLKernelWriter.h index 42d2b07ded..5df148da7b 100644 --- a/compute_kernel_writer/src/cl/CLKernelWriter.h +++ b/compute_kernel_writer/src/cl/CLKernelWriter.h @@ -55,8 +55,12 @@ public: // Misc // ============================================================================================= + /** Similar to @ref KernelWriter::comment() */ void comment(const std::string &text) override; + /** Similar to @ref KernelWriter::op_write_raw_code() */ + void op_write_raw_code(const std::string &raw_code) override; + // ============================================================================================= // Code generation // ============================================================================================= -- cgit v1.2.1