diff options
| -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 // ============================================================================================= |