IVGCVSW-5482 'Add a ClCompileContext parameter to each ClWorkload Constructor'
* Injected CLCompileContext object to each CL workload.
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I4837dbd3d5b56cf743b3b89c944e3cdf8b11a42a
diff --git a/src/backends/cl/workloads/ClTransposeWorkload.cpp b/src/backends/cl/workloads/ClTransposeWorkload.cpp
index b276b22..7ef502e 100644
--- a/src/backends/cl/workloads/ClTransposeWorkload.cpp
+++ b/src/backends/cl/workloads/ClTransposeWorkload.cpp
@@ -27,7 +27,8 @@
}
ClTransposeWorkload::ClTransposeWorkload(const TransposeQueueDescriptor& descriptor,
- const WorkloadInfo& info)
+ const WorkloadInfo& info,
+ const arm_compute::CLCompileContext& clCompileContext)
: BaseWorkload<TransposeQueueDescriptor>(descriptor, info)
{
m_Data.ValidateInputsOutputs(GetName(), 1, 1);
@@ -36,7 +37,9 @@
arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
const armnn::PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings;
// Run the layer.
- m_PermuteFunction.configure(&input, &output,
+ m_PermuteFunction.configure(clCompileContext,
+ &input,
+ &output,
armcomputetensorutils::BuildArmComputeTransposeVector(mappings));
}