From 1df9f6ed4245489b74875893c695367bd0d6e3d8 Mon Sep 17 00:00:00 2001 From: Viet-Hoa Do Date: Mon, 24 Jul 2023 17:57:12 +0100 Subject: Add kernel argument emitting Resolves: COMPMID-6391 Signed-off-by: Viet-Hoa Do Change-Id: I0d54d99ffad275400c6da7fe16deb544553060eb Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10004 Reviewed-by: Anitha Raj Reviewed-by: Gunes Bayir Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins Benchmark: Arm Jenkins --- compute_kernel_writer/include/ckw/Kernel.h | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) (limited to 'compute_kernel_writer/include/ckw/Kernel.h') diff --git a/compute_kernel_writer/include/ckw/Kernel.h b/compute_kernel_writer/include/ckw/Kernel.h index d93ed6f1d3..dc0cad5503 100644 --- a/compute_kernel_writer/include/ckw/Kernel.h +++ b/compute_kernel_writer/include/ckw/Kernel.h @@ -25,7 +25,9 @@ #ifndef CKW_INCLUDE_CKW_KERNEL_H #define CKW_INCLUDE_CKW_KERNEL_H +#include "ckw/KernelArgument.h" #include +#include namespace ckw { @@ -48,22 +50,24 @@ public: /** Initialize a new instance of @ref Kernel class with all emitted kernel information. * * @param[in] language The target language of the kernel. + * @param[in] arguments The list of kernel arguments. * @param[in] source_code The source code of the kernel. */ - Kernel(TargetLanguage language, const std::string &source_code); + Kernel(TargetLanguage language, const std::vector &arguments, const std::string &source_code); /** Get the target language. */ TargetLanguage target_language() const; + /** Get the list of arguments. */ + const std::vector &arguments() const; + /** Get the source code. */ const std::string &source_code() const; - /** Add a tile operand */ - virtual TileOperand &add_operand(const std::string &name, const TileInfo &tile_info) = 0; - private: - TargetLanguage _language; - std::string _source_code; + TargetLanguage _language; + std::vector _arguments; + std::string _source_code; }; } // namespace ckw -- cgit v1.2.1