1 files changed, 5 insertions, 6 deletions
diff --git a/src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwVariableTable.h b/src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwVariableTable.h
index 170fda451b..1c9cb083ea 100644
--- a/src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwVariableTable.h
+++ b/src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwVariableTable.h
@@ -24,13 +24,11 @@
 #ifndef ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_GPUCKWVARIABLETABLE
 #define ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_GPUCKWVARIABLETABLE
 
-#include "acl/AclComponentArgument.h"
+#include "src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwComponentArgument.h"
 #include "arm_compute/core/ITensorInfo.h"
 
 #include <map>
 
-class AclScopedKernelWriter;
-
 namespace arm_compute
 {
 namespace experimental
@@ -38,6 +36,7 @@ namespace experimental
 namespace dynamic_fusion
 {
 class GpuKernelComponentGroup;
+class GpuCkwScopedKernelWriter;
 
 /** A table of all the variables used in the kernel.
  *
@@ -55,12 +54,12 @@ public:
      * @param[in] tensor     Tensor info with which the new variable is associated
      * @param[in] alias      Alias for the variable. Will be used as part of the variable name
      *
-     * @return AclComponentArgument*
+     * @return GpuCkwComponentArgument*
      */
-    AclComponentArgument *declare_variable(const GpuKernelComponentGroup &comp_group, AclScopedKernelWriter &writer, const ITensorInfo *tensor, const std::string &alias = "unnamed");
+    GpuCkwComponentArgument *declare_variable(const GpuKernelComponentGroup &comp_group, GpuCkwScopedKernelWriter &writer, const ITensorInfo *tensor, const std::string &alias = "unnamed");
 
 private:
-    std::map<ITensorInfo::Id, AclComponentArgument> _vars{};
+    std::map<ITensorInfo::Id, GpuCkwComponentArgument> _vars{};
 };
 
 } // namespace dynamic_fusion