24.02.1
|
Go to the documentation of this file.
24 #ifndef SRC_DYNAMIC_FUSION_SKETCH_GPU_GPUKERNELARGUMENT
25 #define SRC_DYNAMIC_FUSION_SKETCH_GPU_GPUKERNELARGUMENT
31 namespace experimental
33 namespace dynamic_fusion
82 : _tensor_info{
tensor_info}, _kernel_arg_info{kernel_arg_info}
88 return _tensor_info.
id();
103 return &_kernel_arg_info;
108 return &_kernel_arg_info;
124 #ifdef ACL_INTERNAL_TEST_CKW_IN_DF
129 enum class TensorStorageType
141 enum class TensorComponentType
191 class GpuKernelArgumentBinding
194 enum class Type : int32_t
199 GpuKernelArgumentBinding(
ITensorInfo::Id id, TensorStorageType storage)
200 : _type{Type::TensorStorage}, _id{
id}, _value{}
202 _value.tensor_storage_type = storage;
204 GpuKernelArgumentBinding(
ITensorInfo::Id id, TensorComponentType component)
205 : _type{Type::TensorComponent}, _id{
id}, _value{}
207 _value.tensor_component_type = component;
211 TensorStorageType tensor_storage_type()
const
214 return _value.tensor_storage_type;
218 TensorComponentType tensor_component_type()
const
221 return _value.tensor_component_type;
241 TensorStorageType tensor_storage_type;
242 TensorComponentType tensor_component_type;
246 #endif // ACL_INTERNAL_TEST_CKW_IN_DF
GpuKernelArgumentInfo(Type type)
Constructor.
@ Image_Export_To_ClImage2D
@ Image_Reinterpret_As_3D
decltype(strategy::transforms) typedef type
Kernel argument information linked with its corresponding ITensorInfo.
ITensorInfo::Id id() const
Get workload tensor id.
GpuKernelArgument(const ITensorInfo &tensor_info, const GpuKernelArgumentInfo &kernel_arg_info)
Constructor.
bool has_valid_id() const
Check if the tensor id is valid.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
GpuKernelArgumentInfo()=default
Default constructor.
GpuKernelArgumentInfo * kernel_argument_info()
Get associated GpuKernelArgumentInfo.
const GpuKernelArgumentInfo * kernel_argument_info() const
Get associated GpuKernelArgumentInfo.
ITensorInfo * tensor_info()
Get associated ITensorInfo.
int32_t Id
An id that uniquely identifies an ITensorInfo within some domain (e.g.
Contain information required to set up a kernel argument at run time.
bool has_valid_id() const
Check if the associated workload tensor has valid id.
const ITensorInfo * tensor_info() const
Get associated ITensorInfo.
Store the tensor's metadata.
Copyright (c) 2017-2024 Arm Limited.
bool operator==(const KernelProperties &config0, const KernelProperties &config1)
ITensorInfo::Id id() const override
Get the workload tensor id of the tensor.
Store the tensor's metadata.
Type
Enumerate all the tensor arguments variants used by all kernel implementations.
@ Image_3D_Export_To_ClImage2D