diff options
Diffstat (limited to 'src/dynamic_fusion/sketch/gpu/ckw_driver/IGpuCkwComponentDriver.h')
-rw-r--r-- | src/dynamic_fusion/sketch/gpu/ckw_driver/IGpuCkwComponentDriver.h | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/dynamic_fusion/sketch/gpu/ckw_driver/IGpuCkwComponentDriver.h b/src/dynamic_fusion/sketch/gpu/ckw_driver/IGpuCkwComponentDriver.h index 15402bc330..62255f1cf6 100644 --- a/src/dynamic_fusion/sketch/gpu/ckw_driver/IGpuCkwComponentDriver.h +++ b/src/dynamic_fusion/sketch/gpu/ckw_driver/IGpuCkwComponentDriver.h @@ -28,8 +28,6 @@ #include "src/dynamic_fusion/sketch/ArgumentPack.h" #include "src/dynamic_fusion/sketch/gpu/components/Types.h" -class AclScopedKernelWriter; - namespace arm_compute { class ITensorInfo; @@ -40,6 +38,7 @@ namespace dynamic_fusion /** Forward declaration */ class GpuKernelComponentGroup; class GpuCkwVariableTable; +class GpuCkwScopedKernelWriter; /** An interface used by @ref GpuCkwDriver to write source code for a kernel component * @@ -90,7 +89,7 @@ public: * * @note @p writer can only be passed via value since the new scope is created in the copy constructor */ - virtual void write_component_code(const ComponentGroup &comp_group, GpuCkwVariableTable &vtable, AclScopedKernelWriter writer) const = 0; + virtual void write_component_code(const ComponentGroup &comp_group, GpuCkwVariableTable &vtable, GpuCkwScopedKernelWriter writer) const = 0; /** Get tensor arguments */ ArgumentPack<ITensorInfo> tensors() const { |