diff options
Diffstat (limited to 'src/dynamic_fusion/sketch/gpu/components/cl/ClComponentCast.h')
-rw-r--r-- | src/dynamic_fusion/sketch/gpu/components/cl/ClComponentCast.h | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/dynamic_fusion/sketch/gpu/components/cl/ClComponentCast.h b/src/dynamic_fusion/sketch/gpu/components/cl/ClComponentCast.h index 84d6f07f16..174f9670b3 100644 --- a/src/dynamic_fusion/sketch/gpu/components/cl/ClComponentCast.h +++ b/src/dynamic_fusion/sketch/gpu/components/cl/ClComponentCast.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2022 Arm Limited. + * Copyright (c) 2022-2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -49,6 +49,7 @@ private: /** Forward declaration */ class ClTemplateCast; +class GpuCkwCast; class ClComponentCast final : public IGpuKernelComponent { @@ -117,6 +118,8 @@ public: ClComponentCast &operator=(ClComponentCast &&component) = default; /** Get template writer for the component */ const IGpuTemplateComponentWriter *template_writer() const override; + /** Get GPU kernel writer for the component */ + const IGpuCkwComponentDriver *ckw_component_driver() const override; /** Get component type */ GpuComponentType type() const override { @@ -125,6 +128,7 @@ public: private: std::unique_ptr<ClTemplateCast> _component_writer; + std::unique_ptr<GpuCkwCast> _ckw_driver; }; } // namespace dynamic_fusion } // namespace experimental |