24.02.1
|
Hold the generated kernel source code and other information required to compile and run the workload. More...
#include <GpuWorkloadSourceCode.h>
Public Member Functions | |
GpuWorkloadSourceCode ()=default | |
Default constructor. More... | |
UnitWorkloadId | add_unit_workload (const GpuKernelSourceCode &kernel_code, const UnitWorkloadStage &stage, const MemoryDescriptorMap &mem_map, const GpuWorkloadContext *context) |
Add a unit workload to the workload code. More... | |
const GpuUnitWorkload & | query_unit_workload (UnitWorkloadId id) const |
Get a unit workload from its id. More... | |
std::vector< UnitWorkloadId > | unit_workloads () const |
Get all unit workloads sorted in topological order. More... | |
const GpuWorkloadArgument * | query_tensor (ITensorInfo::Id t_id) const |
Get a GpuWorkloadArgument from its associated tensor id. More... | |
std::vector< ITensorInfo::Id > | tensors () const |
Get all tensors in the entire workload. More... | |
std::vector< UnitWorkloadId > | get_unit_workloads_from_tensor (ITensorInfo::Id t_id) const |
Get all unit workloads connected to the tensor with t_id . More... | |
Hold the generated kernel source code and other information required to compile and run the workload.
Definition at line 239 of file GpuWorkloadSourceCode.h.
|
default |
Default constructor.
|
inline |
Add a unit workload to the workload code.
[in] | kernel_code | GpuKernelSourceCode to be contained within the unit workload |
[in] | stage | Stage of the unit workload |
[in] | mem_map | MemoryDescriptor map for all tensors within the unit workload |
[in] | context | GpuWorkloadContext associated with the unit workload |
Definition at line 253 of file GpuWorkloadSourceCode.h.
References GpuKernelSourceCode::arguments(), ARM_COMPUTE_UNUSED, arm_compute::test::validation::context, and GpuWorkloadArgument::tensor_info().
Referenced by GpuKernelComponentStream::write_workload_code().
|
inline |
Get all unit workloads connected to the tensor with t_id
.
Definition at line 335 of file GpuWorkloadSourceCode.h.
|
inline |
Get a GpuWorkloadArgument from its associated tensor id.
Definition at line 320 of file GpuWorkloadSourceCode.h.
|
inline |
Get a unit workload from its id.
Definition at line 303 of file GpuWorkloadSourceCode.h.
References ARM_COMPUTE_ERROR_ON.
|
inline |
Get all tensors in the entire workload.
Definition at line 325 of file GpuWorkloadSourceCode.h.
|
inline |
Get all unit workloads sorted in topological order.
Definition at line 309 of file GpuWorkloadSourceCode.h.