diff options
Diffstat (limited to 'src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwDriver.h')
-rw-r--r-- | src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwDriver.h | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwDriver.h b/src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwDriver.h index 2084b72098..c6e03f6e03 100644 --- a/src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwDriver.h +++ b/src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwDriver.h @@ -28,6 +28,8 @@ #include "src/dynamic_fusion/sketch/gpu/GpuKernelComponentGroup.h" #include "src/dynamic_fusion/sketch/gpu/IGpuKernelWriter.h" +#include "ckw/Kernel.h" + #include <map> #include <string> @@ -66,11 +68,12 @@ public: std::string get_config_id() override; /** Generate execution window */ Window get_window() const override; - /** Get the kernel argument lists of the kernel*/ - std::map<ITensorInfo::Id, GpuKernelArgument> get_tensors() override; + /** Get the flat list of arguments of the kernel*/ + GpuKernelArgumentList get_kernel_arguments() override; private: GpuKernelComponentGroup _components{}; + ckw::Kernel _kernel; }; } // namespace dynamic_fusion |