24.04
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_COMPONENTS_GPUCKWCAST
25 #define ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_COMPONENTS_GPUCKWCAST
33 namespace experimental
35 namespace dynamic_fusion
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(GpuCkwCast)
GpuCkwCast(ComponentId id, const ArgumentPack< ITensorInfo > &tensors, const Attributes &attributes)
Constructor.
Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator.
Window get_window() const override
Generate the execution window for the component.
This is a generic class that packs the arguments of an operator.
~GpuCkwCast() override=default
Destructor.
A table of all the variables used in the kernel.
CastAttributes Attributes
Attributes are a set of backend-agnostic parameters that define what a component does.
virtual void write_component_code(const ComponentGroup &comp_group, GpuCkwVariableTable &vtable, GpuCkwScopedKernelWriter writer) const override
Generate kernel component code.
An interface used by GpuCkwDriver to write source code for a kernel component.
int32_t ComponentId
Uniquely identifies a kernel component within a workload.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
ArgumentPack< ITensorInfo > tensors() const
Get tensor arguments.
A group of gpu kernel components to be fused together PRECONDITIONS:
Store the tensor's metadata.
Helper to automatically manage kernel writer ID space.