24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_IGPUCKWCOMPONENTDRIVER
25 #define ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_IGPUCKWCOMPONENTDRIVER
35 namespace experimental
37 namespace dynamic_fusion
40 class GpuKernelComponentGroup;
41 class GpuCkwVariableTable;
42 class GpuCkwScopedKernelWriter;
This is a generic class that packs the arguments of an operator.
ComponentId id() const
Get component id.
A table of all the variables used in the kernel.
IGpuCkwComponentDriver(ComponentId id, const ArgumentPack< ITensorInfo > &tensors)
Constructor.
virtual void write_component_code(const ComponentGroup &comp_group, GpuCkwVariableTable &vtable, GpuCkwScopedKernelWriter writer) const =0
Generate kernel component code.
An interface used by GpuCkwDriver to write source code for a kernel component.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
virtual std::string get_name(const ComponentGroup &comp_group) const
Generate the name of the component.
virtual ~IGpuCkwComponentDriver()
Destructor.
int32_t ComponentId
Uniquely identifies a kernel component within a workload.
virtual Window get_window() const
Generate the execution window for the component.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
virtual std::string get_tuner_id(const ComponentGroup &comp_group) const
Generate the tuner id of the component This id should capture all the parameters that distinguish one...
ArgumentPack< ITensorInfo > tensors() const
Get tensor arguments.
A group of gpu kernel components to be fused together PRECONDITIONS:
GpuKernelComponentGroup ComponentGroup
Helper to automatically manage kernel writer ID space.