diff options
| author | Gunes Bayir <gunes.bayir@arm.com> | 2023-07-27 22:52:32 +0100 |
|---|---|---|
| committer | Gunes Bayir <gunes.bayir@arm.com> | 2023-08-02 15:27:41 +0000 |
| commit | 366514d54fd43321f440ebddc49fda28d40b8e49 (patch) | |
| tree | 2b307b2663066aee02b235e6d93cc7d96d29d457 | |
| parent | 4cb0bd488f70a07b222e1ed7008d888642dfec6f (diff) | |
| download | ComputeLibrary-366514d54fd43321f440ebddc49fda28d40b8e49.tar.gz | |
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 <gunes.bayir@arm.com>
Change-Id: I04288457ac1871aea8bf10276bf9f078e17fc4b4
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9996
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
| -rw-r--r-- | compute_kernel_writer/include/ckw/KernelWriter.h | 9 | ||||
| -rw-r--r-- | compute_kernel_writer/src/cl/CLKernelWriter.cpp | 5 | ||||
| -rw-r--r-- | compute_kernel_writer/src/cl/CLKernelWriter.h | 4 |
3 files changed, 18 insertions, 0 deletions
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 // ============================================================================================= |