diff options
Diffstat (limited to 'src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwVariableTable.h')
-rw-r--r-- | src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwVariableTable.h | 11 |
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 |