24.04
|
#include <GpuCkwStore.h>
Public Member Functions | |
GpuCkwStore (ComponentId id, const ArgumentPack< ITensorInfo > &tensors) | |
Constructor. More... | |
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE (GpuCkwStore) | |
~GpuCkwStore () override=default | |
Destructor. More... | |
virtual void | write_component_code (const ComponentGroup &comp_group, GpuCkwVariableTable &vtable, GpuCkwScopedKernelWriter writer) const override |
Generate kernel component code. More... | |
std::string | get_name (const ComponentGroup &comp_group) const override |
Generate the name of the component. More... | |
Public Member Functions inherited from IGpuCkwComponentDriver | |
IGpuCkwComponentDriver (ComponentId id, const ArgumentPack< ITensorInfo > &tensors) | |
Constructor. More... | |
virtual | ~IGpuCkwComponentDriver () |
Destructor. More... | |
ArgumentPack< ITensorInfo > | tensors () const |
Get tensor arguments. More... | |
virtual Window | get_window () const |
Generate the execution window for the component. More... | |
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 kernel's lws tuning from another. More... | |
ComponentId | id () const |
Get component id. More... | |
Additional Inherited Members | |
Public Types inherited from IGpuCkwComponentDriver | |
using | ComponentGroup = GpuKernelComponentGroup |
Definition at line 36 of file GpuCkwStore.h.
GpuCkwStore | ( | ComponentId | id, |
const ArgumentPack< ITensorInfo > & | tensors | ||
) |
Constructor.
[in] | id | Component id |
[in] | tensors | Tensor arguments to the component |
Definition at line 41 of file GpuCkwStore.cpp.
References IGpuCkwComponentDriver::id(), and IGpuCkwComponentDriver::tensors().
|
overridedefault |
Destructor.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE | ( | GpuCkwStore | ) |
|
overridevirtual |
Generate the name of the component.
This will be concatenated with other components' names to form the name of the kernel
Reimplemented from IGpuCkwComponentDriver.
Definition at line 137 of file GpuCkwStore.cpp.
References ARM_COMPUTE_UNUSED.
|
overridevirtual |
Generate kernel component code.
[in] | comp_group | Component group of which the component is a part of |
[in,out] | vtable | Table of variables declared by each component |
[in,out] | writer | CKW writer that writes code scoped to this kernel component. @note @p writer can only be passed via value since the new scope is created in the copy constructor |
Implements IGpuCkwComponentDriver.
Definition at line 47 of file GpuCkwStore.cpp.
References GpuCkwVariableTable::declare_variable(), ITensorInfo::dimension(), arm_compute::test::validation::dst, arm_compute::experimental::dynamic_fusion::get_coordinate_from_gws(), arm_compute::experimental::dynamic_fusion::get_coordinate_from_gws_overlapping_min(), and arm_compute::test::validation::src.