blob: 6bf510a2efee9e86d51cfaa5f14957deeedf874d [file] [log] [blame]
Laurent Carlier749294b2020-06-01 09:03:17 +01001//
Teresa Charlin8398edc2020-07-20 14:23:02 +01002// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +01003// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +00004//
5#include "ClWorkloadFactory.hpp"
David Beck79141b92018-10-23 16:09:36 +01006#include "ClBackendId.hpp"
Sadik Armagan04a72972020-09-14 15:44:18 +01007#include "ClBackendModelContext.hpp"
Matthew Sloyan80fbcd52021-01-07 13:28:47 +00008#include "ClContextDeserializer.hpp"
9#include "ClContextSerializer.hpp"
telsoa014fcda012018-03-09 14:13:49 +000010
Aron Virginas-Tar5caf9072018-11-14 18:35:18 +000011#include <Layer.hpp>
12
David Beck0dbe0ee2018-09-24 15:59:27 +010013#include <armnn/Exceptions.hpp>
Sadik Armagan6b9eba22021-11-01 07:27:50 +000014#include <armnn/Logging.hpp>
David Beck0dbe0ee2018-09-24 15:59:27 +010015#include <armnn/Utils.hpp>
Jan Eilers8eb25602020-03-09 12:13:48 +000016#include <armnn/utility/IgnoreUnused.hpp>
Matthew Sloyan171214c2020-09-09 09:07:37 +010017#include <armnn/utility/NumericCast.hpp>
Jan Eilersbb446e52020-04-02 13:56:54 +010018#include <armnn/utility/PolymorphicDowncast.hpp>
telsoa014fcda012018-03-09 14:13:49 +000019
Aron Virginas-Tar5caf9072018-11-14 18:35:18 +000020#include <backendsCommon/MakeWorkloadHelper.hpp>
Colm Donelan0c479742021-12-10 12:43:54 +000021#include <armnn/backends/MemCopyWorkload.hpp>
Derek Lambertif674aa02019-08-01 15:56:25 +010022#include <backendsCommon/MemImportWorkload.hpp>
Colm Donelan0c479742021-12-10 12:43:54 +000023#include <armnn/backends/TensorHandle.hpp>
telsoa01c577f2c2018-08-31 09:22:23 +010024
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +000025#include <cl/ClTensorHandle.hpp>
26#include <cl/workloads/ClWorkloads.hpp>
27#include <cl/workloads/ClWorkloadUtils.hpp>
telsoa014fcda012018-03-09 14:13:49 +000028
Aron Virginas-Tar5caf9072018-11-14 18:35:18 +000029#include <arm_compute/core/CL/CLKernelLibrary.h>
30#include <arm_compute/runtime/CL/CLBufferAllocator.h>
31#include <arm_compute/runtime/CL/CLScheduler.h>
telsoa014fcda012018-03-09 14:13:49 +000032
Rob Hughes9542f902021-07-14 09:48:54 +010033#include <armnnUtils/Filesystem.hpp>
Matthew Sloyan80fbcd52021-01-07 13:28:47 +000034#include <fstream>
Sadik Armagandea8fb62020-11-26 10:38:11 +000035
Sadik Armaganb7851f92021-10-06 16:37:02 +010036#include <sys/stat.h>
37
telsoa014fcda012018-03-09 14:13:49 +000038namespace armnn
39{
40
David Beck79141b92018-10-23 16:09:36 +010041namespace
42{
43static const BackendId s_Id{ClBackendId()};
44}
45
telsoa01c577f2c2018-08-31 09:22:23 +010046bool ClWorkloadFactory::IsLayerSupported(const Layer& layer,
David Beck29c75de2018-10-23 13:35:58 +010047 Optional<DataType> dataType,
telsoa01c577f2c2018-08-31 09:22:23 +010048 std::string& outReasonIfUnsupported)
telsoa014fcda012018-03-09 14:13:49 +000049{
David Beck79141b92018-10-23 16:09:36 +010050 return IWorkloadFactory::IsLayerSupported(s_Id, layer, dataType, outReasonIfUnsupported);
51}
52
Sadik Armagan04a72972020-09-14 15:44:18 +010053bool ClWorkloadFactory::IsLayerSupported(const IConnectableLayer& layer,
54 Optional<DataType> dataType,
55 std::string& outReasonIfUnsupported,
56 const ModelOptions& modelOptions)
57{
58 return IWorkloadFactory::IsLayerSupported(s_Id, layer, dataType, outReasonIfUnsupported, modelOptions);
59}
60
David Beck79141b92018-10-23 16:09:36 +010061const BackendId& ClWorkloadFactory::GetBackendId() const
62{
63 return s_Id;
telsoa014fcda012018-03-09 14:13:49 +000064}
65
Sadik Armagandea8fb62020-11-26 10:38:11 +000066void ClWorkloadFactory::AfterWorkloadsCreated()
67{
68 if(m_ModelContextPtr)
69 {
70 auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
71 if (modelOptions->SaveCachedNetwork())
72 {
Sadik Armaganb7851f92021-10-06 16:37:02 +010073 ClContextSerializer serializer;
74 serializer.Serialize(m_CLCompileContext);
75 auto cachedFd = modelOptions->GetCachedFileDescriptor();
76 if (cachedFd != -1)
77 {
78 std::vector<uint8_t> compiledContextData;
79 std::stringstream stream;
80 bool serialized = serializer.SaveSerializedToStream(stream);
81 if (serialized)
82 {
83 std::string const serializedString{stream.str()};
84 std::copy(serializedString.begin(),
85 serializedString.end(),
86 std::back_inserter(compiledContextData));
Sadik Armagan6b9eba22021-11-01 07:27:50 +000087 auto success = write(cachedFd, compiledContextData.data(), compiledContextData.size());
88 if (success == -1)
89 {
90 ARMNN_LOG(info) << "ClWorkloadFactory:: Could not cache the compiled context!";
91 }
Sadik Armaganb7851f92021-10-06 16:37:02 +010092 }
93 }
94
Sadik Armagandea8fb62020-11-26 10:38:11 +000095 // Save map to a filepath provided in ModelOptions
96 auto filePath = modelOptions->GetCachedNetworkFilePath();
97 if (filePath != "" && fs::exists(filePath) && fs::is_regular_file(filePath))
98 {
Matthew Sloyan80fbcd52021-01-07 13:28:47 +000099 // Serialize ClContext to the file specified
Matthew Sloyan80fbcd52021-01-07 13:28:47 +0000100 std::ofstream file(filePath, std::ios::out | std::ios::binary);
101 serializer.SaveSerializedToStream(file);
Sadik Armagandea8fb62020-11-26 10:38:11 +0000102 }
103 }
104 }
105}
106
Aron Virginas-Tara8e06ed2018-10-19 16:46:15 +0100107template <typename FloatWorkload, typename Uint8Workload, typename QueueDescriptorType, typename... Args>
108std::unique_ptr<IWorkload> ClWorkloadFactory::MakeWorkload(const QueueDescriptorType& descriptor,
109 const WorkloadInfo& info,
110 Args&&... args)
111{
112 try
113 {
114 return MakeWorkloadHelper<FloatWorkload, Uint8Workload>(descriptor, info, std::forward<Args>(args)...);
115 }
116 catch (const cl::Error& clError)
117 {
118 throw WrapClError(clError, CHECK_LOCATION());
119 }
120}
121
122template <typename Workload, typename QueueDescriptorType, typename... Args>
123std::unique_ptr<IWorkload> ClWorkloadFactory::MakeWorkload(const QueueDescriptorType& descriptor,
124 const WorkloadInfo& info,
125 Args&&... args)
126{
127 try
128 {
129 return std::make_unique<Workload>(descriptor, info, std::forward<Args>(args)...);
130 }
131 catch (const cl::Error& clError)
132 {
133 throw WrapClError(clError, CHECK_LOCATION());
134 }
135}
136
Sadik Armagandea8fb62020-11-26 10:38:11 +0000137void ClWorkloadFactory::InitializeCLCompileContext()
138{
139 // Initialize our m_CLCompileContext using default device and context
Sadik Armagane9444752020-12-02 11:28:58 +0000140 auto context = arm_compute::CLKernelLibrary::get().context();
141 auto device = arm_compute::CLKernelLibrary::get().get_device();
Sadik Armagandea8fb62020-11-26 10:38:11 +0000142 m_CLCompileContext = arm_compute::CLCompileContext(context, device);
143
144 if (m_ModelContextPtr)
145 {
146 // Load saved programs if the user has set a filepath
147 auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
148 auto filePath = modelOptions->GetCachedNetworkFilePath();
Sadik Armaganb7851f92021-10-06 16:37:02 +0100149 if (!(modelOptions->SaveCachedNetwork()))
Sadik Armagandea8fb62020-11-26 10:38:11 +0000150 {
Matthew Sloyan80fbcd52021-01-07 13:28:47 +0000151 ClContextDeserializer deserializer;
Sadik Armaganb7851f92021-10-06 16:37:02 +0100152 auto cachedFd = modelOptions->GetCachedFileDescriptor();
153 if (cachedFd != -1)
154 {
155 struct stat statBuffer;
156 if (fstat(cachedFd, &statBuffer) == 0)
157 {
158 long dataSize = static_cast<long>(statBuffer.st_size);
159 if( dataSize > 0)
160 {
161 auto offset = lseek(cachedFd, 0, SEEK_CUR);
162 if (offset == 0)
163 {
164 std::vector <uint8_t> compiledContextData(static_cast<unsigned int>(dataSize));
Sadik Armagan6b9eba22021-11-01 07:27:50 +0000165 auto success = pread(cachedFd, compiledContextData.data(), compiledContextData.size(), 0);
166 if (success != -1)
167 {
168 deserializer.DeserializeFromBinary(m_CLCompileContext,
169 context,
170 device,
171 compiledContextData);
172 }
Sadik Armaganb7851f92021-10-06 16:37:02 +0100173 }
174 }
175
176 }
177 }
178
179 if (filePath != "" && fs::exists(filePath) && fs::is_regular_file(filePath))
180 {
181 // Deserialize binary file and load into m_CLCompileContext
182 deserializer.Deserialize(m_CLCompileContext, context, device, filePath);
183 }
Sadik Armagandea8fb62020-11-26 10:38:11 +0000184 }
185 }
186}
187
Aron Virginas-Tar56055192018-11-12 18:10:43 +0000188ClWorkloadFactory::ClWorkloadFactory(const std::shared_ptr<ClMemoryManager>& memoryManager)
Sadik Armagan04a72972020-09-14 15:44:18 +0100189 : m_MemoryManager(memoryManager), m_ModelContextPtr(IBackendInternal::IBackendSpecificModelContextPtr{})
190{
Sadik Armagandea8fb62020-11-26 10:38:11 +0000191 InitializeCLCompileContext();
Sadik Armagan04a72972020-09-14 15:44:18 +0100192}
193
194ClWorkloadFactory::ClWorkloadFactory(const std::shared_ptr<ClMemoryManager>& memoryManager,
195 const IBackendInternal::IBackendSpecificModelContextPtr& modelContextPtr)
196 : m_MemoryManager(memoryManager), m_ModelContextPtr(modelContextPtr)
telsoa014fcda012018-03-09 14:13:49 +0000197{
Sadik Armagandea8fb62020-11-26 10:38:11 +0000198 InitializeCLCompileContext();
telsoa014fcda012018-03-09 14:13:49 +0000199}
200
David Monahan3fb7e102019-08-20 11:25:29 +0100201std::unique_ptr<ITensorHandle> ClWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo,
202 const bool IsMemoryManaged) const
telsoa014fcda012018-03-09 14:13:49 +0000203{
Jan Eilers8eb25602020-03-09 12:13:48 +0000204 IgnoreUnused(IsMemoryManaged);
telsoa01c577f2c2018-08-31 09:22:23 +0100205 std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo);
Aron Virginas-Tar56055192018-11-12 18:10:43 +0000206 tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());
telsoa01c577f2c2018-08-31 09:22:23 +0100207
208 return tensorHandle;
telsoa014fcda012018-03-09 14:13:49 +0000209}
210
Francis Murtagh351d13d2018-09-24 15:01:18 +0100211std::unique_ptr<ITensorHandle> ClWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo,
David Monahan3fb7e102019-08-20 11:25:29 +0100212 DataLayout dataLayout,
213 const bool IsMemoryManaged) const
Francis Murtagh351d13d2018-09-24 15:01:18 +0100214{
Jan Eilers8eb25602020-03-09 12:13:48 +0000215 IgnoreUnused(IsMemoryManaged);
Francis Murtagh351d13d2018-09-24 15:01:18 +0100216 std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo, dataLayout);
Aron Virginas-Tar56055192018-11-12 18:10:43 +0000217 tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());
Francis Murtagh351d13d2018-09-24 15:01:18 +0100218
219 return tensorHandle;
220}
221
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100222std::unique_ptr<ITensorHandle> ClWorkloadFactory::CreateSubTensorHandle(ITensorHandle& parent,
223 TensorShape const& subTensorShape,
telsoa014fcda012018-03-09 14:13:49 +0000224 unsigned int const* subTensorOrigin) const
225{
telsoa014fcda012018-03-09 14:13:49 +0000226 arm_compute::Coordinates coords;
227 arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape);
228
229 coords.set_num_dimensions(subTensorShape.GetNumDimensions());
230 for (unsigned int i = 0; i < subTensorShape.GetNumDimensions(); i++)
231 {
telsoa01c577f2c2018-08-31 09:22:23 +0100232 // Arm compute indexes tensor coords in reverse order.
telsoa014fcda012018-03-09 14:13:49 +0000233 unsigned int revertedIndex = subTensorShape.GetNumDimensions() - i - 1;
Matthew Sloyan171214c2020-09-09 09:07:37 +0100234 coords.set(i, armnn::numeric_cast<int>(subTensorOrigin[revertedIndex]));
telsoa014fcda012018-03-09 14:13:49 +0000235 }
236
Derek Lamberti0790dce2019-04-15 18:37:35 +0100237 const arm_compute::TensorShape parentShape = armcomputetensorutils::BuildArmComputeTensorShape(parent.GetShape());
238 if (!::arm_compute::error_on_invalid_subtensor(__func__, __FILE__, __LINE__, parentShape, coords, shape))
239 {
240 return nullptr;
241 }
242
telsoa01c577f2c2018-08-31 09:22:23 +0100243 return std::make_unique<ClSubTensorHandle>(
Jan Eilersbb446e52020-04-02 13:56:54 +0100244 PolymorphicDowncast<IClTensorHandle*>(&parent), shape, coords);
telsoa014fcda012018-03-09 14:13:49 +0000245}
246
Teresa Charlin611c7fb2022-01-07 09:47:29 +0000247std::unique_ptr<IWorkload> ClWorkloadFactory::CreateWorkload(LayerType type,
248 const QueueDescriptor& descriptor,
249 const WorkloadInfo& info) const
250{
251 switch(type)
252 {
253 case LayerType::Activation :
254 {
255 auto activationQueueDescriptor = PolymorphicDowncast<const ActivationQueueDescriptor*>(&descriptor);
256 return MakeWorkload<ClActivationWorkload>(*activationQueueDescriptor, info, m_CLCompileContext);
257 }
258 case LayerType::Addition :
259 {
260 auto additionQueueDescriptor = PolymorphicDowncast<const AdditionQueueDescriptor*>(&descriptor);
261 return MakeWorkload<ClAdditionWorkload>(*additionQueueDescriptor, info, m_CLCompileContext);
262 }
263 case LayerType::ArgMinMax :
264 {
265 auto argMinMaxQueueDescriptor = PolymorphicDowncast<const ArgMinMaxQueueDescriptor*>(&descriptor);
Cathal Corbettea660e12022-06-08 17:36:13 +0100266 return MakeWorkload<ClArgMinMaxWorkload>(*argMinMaxQueueDescriptor, info, m_CLCompileContext);
Teresa Charlin611c7fb2022-01-07 09:47:29 +0000267 }
Teresa Charlin94916a52022-10-19 08:48:07 +0100268 case LayerType::BatchMatMul :
269 {
270 auto batchMatMulQueueDescriptor = PolymorphicDowncast<const BatchMatMulQueueDescriptor*>(&descriptor);
271 return std::make_unique<ClBatchMatMulWorkload>(*batchMatMulQueueDescriptor, info, m_CLCompileContext);
272 }
Teresa Charlin611c7fb2022-01-07 09:47:29 +0000273 case LayerType::BatchNormalization :
274 {
275 auto batchNormalizationQueueDescriptor
276 = PolymorphicDowncast<const BatchNormalizationQueueDescriptor*>(&descriptor);
277 return MakeWorkload<ClBatchNormalizationFloatWorkload, NullWorkload>
278 (*batchNormalizationQueueDescriptor, info, m_CLCompileContext);
279 }
280 case LayerType::BatchToSpaceNd :
281 {
282 auto batchToSpaceNdQueueDescriptor
283 = PolymorphicDowncast<const BatchToSpaceNdQueueDescriptor*>(&descriptor);
284 return MakeWorkload<ClBatchToSpaceNdWorkload>(*batchToSpaceNdQueueDescriptor, info, m_CLCompileContext);
285 }
286 case LayerType::Cast :
287 {
288 auto castQueueDescriptor = PolymorphicDowncast<const CastQueueDescriptor*>(&descriptor);
289 return MakeWorkload<ClCastWorkload>(*castQueueDescriptor, info, m_CLCompileContext);
290 }
291 case LayerType::ChannelShuffle :
292 {
293 auto channelShuffleQueueDescriptor
294 = PolymorphicDowncast<const ChannelShuffleQueueDescriptor*>(&descriptor);
295 return MakeWorkload<ClChannelShuffleWorkload>(*channelShuffleQueueDescriptor, info, m_CLCompileContext);
296 }
297 case LayerType::Comparison :
298 {
299 auto comparisonQueueDescriptor = PolymorphicDowncast<const ComparisonQueueDescriptor*>(&descriptor);
300 return MakeWorkload<ClComparisonWorkload>(*comparisonQueueDescriptor, info, m_CLCompileContext);
301 }
302 case LayerType::Concat :
303 {
304 auto concatQueueDescriptor = PolymorphicDowncast<const ConcatQueueDescriptor*>(&descriptor);
305 return MakeWorkload<ClConcatWorkload>(*concatQueueDescriptor, info, m_CLCompileContext);
306 }
307 case LayerType::Constant :
308 {
309 auto constantQueueDescriptor = PolymorphicDowncast<const ConstantQueueDescriptor*>(&descriptor);
310 return MakeWorkload<ClConstantWorkload>(*constantQueueDescriptor, info, m_CLCompileContext);
311 }
312 case LayerType::ConvertFp16ToFp32 :
313 {
314 auto convertFp16ToFp32QueueDescriptor
315 = PolymorphicDowncast<const ConvertFp16ToFp32QueueDescriptor*>(&descriptor);
316 return MakeWorkload<ClConvertFp16ToFp32Workload>(*convertFp16ToFp32QueueDescriptor,
317 info,
318 m_CLCompileContext);
319 }
320 case LayerType::ConvertFp32ToFp16 :
321 {
322 auto convertFp32ToFp16QueueDescriptor
323 = PolymorphicDowncast<const ConvertFp32ToFp16QueueDescriptor*>(&descriptor);
324 return MakeWorkload<ClConvertFp32ToFp16Workload>(*convertFp32ToFp16QueueDescriptor,
325 info,
326 m_CLCompileContext);
327 }
328 case LayerType::Convolution2d :
329 {
330 auto convolution2dQueueDescriptor = PolymorphicDowncast<const Convolution2dQueueDescriptor*>(&descriptor);
331
332 bool isFastMathEnabled = false;
333 if (m_ModelContextPtr)
334 {
335 if (m_ModelContextPtr.get() != nullptr)
336 {
337 auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
338 if (modelOptions)
339 {
340 isFastMathEnabled = modelOptions->IsFastMathEnabled();
341 }
342 }
343 }
344 return MakeWorkload<ClConvolution2dWorkload>(*convolution2dQueueDescriptor,
345 info,
346 m_MemoryManager->GetIntraLayerManager(),
347 m_CLCompileContext,
348 isFastMathEnabled);
349 }
350 case LayerType::Convolution3d :
351 {
352 auto convolution3dQueueDescriptor = PolymorphicDowncast<const Convolution3dQueueDescriptor*>(&descriptor);
353
354 bool isFastMathEnabled = false;
355 if (m_ModelContextPtr)
356 {
357 if (m_ModelContextPtr.get() != nullptr)
358 {
359 auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
360 if (modelOptions)
361 {
362 isFastMathEnabled = modelOptions->IsFastMathEnabled();
363 }
364 }
365 }
366 return MakeWorkload<ClConvolution3dWorkload>(*convolution3dQueueDescriptor,
367 info,
368 m_MemoryManager->GetIntraLayerManager(),
369 m_CLCompileContext,
370 isFastMathEnabled);
371 }
372 case LayerType::Debug :
373 {
374 auto debugQueueDescriptor = PolymorphicDowncast<const DebugQueueDescriptor*>(&descriptor);
375 return MakeWorkload<NullWorkload, NullWorkload>(*debugQueueDescriptor, info, m_CLCompileContext);
376 }
377 case LayerType::DepthToSpace :
378 {
379 auto depthToSpaceQueueDescriptor = PolymorphicDowncast<const DepthToSpaceQueueDescriptor*>(&descriptor);
380 return MakeWorkload<ClDepthToSpaceWorkload>(*depthToSpaceQueueDescriptor, info, m_CLCompileContext);
381 }
382 case LayerType::DepthwiseConvolution2d :
383 {
384 auto depthwiseConvolution2dQueueDescriptor
385 = PolymorphicDowncast<const DepthwiseConvolution2dQueueDescriptor*>(&descriptor);
386 return MakeWorkload<ClDepthwiseConvolutionWorkload>(*depthwiseConvolution2dQueueDescriptor,
387 info,
388 m_CLCompileContext);
389 }
390 case LayerType::Dequantize :
391 {
392 auto dequantizeQueueDescriptor = PolymorphicDowncast<const DequantizeQueueDescriptor*>(&descriptor);
393 return MakeWorkload<ClDequantizeWorkload>(*dequantizeQueueDescriptor, info, m_CLCompileContext);
394 }
395 case LayerType::DetectionPostProcess :
396 {
397 auto detectionPostProcessQueueDescriptor
398 = PolymorphicDowncast<const DetectionPostProcessQueueDescriptor*>(&descriptor);
399 return MakeWorkload<NullWorkload, NullWorkload>(*detectionPostProcessQueueDescriptor,
400 info,
401 m_CLCompileContext);
402 }
403 case LayerType::Division :
404 {
405 auto divisionQueueDescriptor = PolymorphicDowncast<const DivisionQueueDescriptor*>(&descriptor);
406 return std::make_unique<ClDivisionWorkload>(*divisionQueueDescriptor, info, m_CLCompileContext);
407 }
408 case LayerType::ElementwiseUnary :
409 {
410 auto elementwiseUnaryQueueDescriptor
411 = PolymorphicDowncast<const ElementwiseUnaryQueueDescriptor*>(&descriptor);
412
413 switch(elementwiseUnaryQueueDescriptor->m_Parameters.m_Operation)
414 {
415 case UnaryOperation::Abs:
416 {
417 AbsQueueDescriptor absQueueDescriptor;
418 absQueueDescriptor.m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs;
419 absQueueDescriptor.m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs;
420
421 return std::make_unique<ClAbsWorkload>(absQueueDescriptor, info, m_CLCompileContext);
422 }
423 case UnaryOperation::Exp:
424 return std::make_unique<ClExpWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext);
425 case UnaryOperation::Log:
426 return std::make_unique<ClLogWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext);
427 case UnaryOperation::LogicalNot:
428 return std::make_unique<ClLogicalNotWorkload>(*elementwiseUnaryQueueDescriptor,
429 info,
430 m_CLCompileContext);
431 case UnaryOperation::Neg:
432 return std::make_unique<ClNegWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext);
433 case UnaryOperation::Rsqrt:
434 {
435 RsqrtQueueDescriptor rsqrtQueueDescriptor;
436 rsqrtQueueDescriptor.m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs;
437 rsqrtQueueDescriptor.m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs;
438
439 return std::make_unique<ClRsqrtWorkload>(rsqrtQueueDescriptor, info, m_CLCompileContext);
440 }
441 case UnaryOperation::Sin:
442 return std::make_unique<ClSinWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext);
Teresa Charlin03027232022-05-09 17:27:08 +0100443 case UnaryOperation::Sqrt:
444 return std::make_unique<ClSqrtWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext);
Teresa Charlin611c7fb2022-01-07 09:47:29 +0000445 default:
446 return nullptr;
447 }
448 }
449 case LayerType::Fill :
450 {
451 auto fillQueueDescriptor = PolymorphicDowncast<const FillQueueDescriptor*>(&descriptor);
452 return std::make_unique<ClFillWorkload>(*fillQueueDescriptor, info, m_CLCompileContext);
453 }
454 case LayerType::Floor :
455 {
456 auto floorQueueDescriptor = PolymorphicDowncast<const FloorQueueDescriptor*>(&descriptor);
457 return MakeWorkload<ClFloorFloatWorkload, NullWorkload>(*floorQueueDescriptor, info, m_CLCompileContext);
458 }
459 case LayerType::FullyConnected :
460 {
461 auto fullyConnectedQueueDescriptor
462 = PolymorphicDowncast<const FullyConnectedQueueDescriptor*>(&descriptor);
463 return MakeWorkload<ClFullyConnectedWorkload>(*fullyConnectedQueueDescriptor,
464 info,
465 m_MemoryManager->GetIntraLayerManager(),
466 m_CLCompileContext);
467 }
468 case LayerType::Gather :
469 {
470 auto gatherQueueDescriptor = PolymorphicDowncast<const GatherQueueDescriptor*>(&descriptor);
471 return MakeWorkload<ClGatherWorkload>(*gatherQueueDescriptor, info, m_CLCompileContext);
472 }
Teresa Charlin989e2f62022-04-27 16:26:11 +0100473 case LayerType::GatherNd :
474 {
475 auto gatherNdQueueDescriptor = PolymorphicDowncast<const GatherNdQueueDescriptor*>(&descriptor);
476 return MakeWorkload<ClGatherNdWorkload>(*gatherNdQueueDescriptor, info, m_CLCompileContext);
477 }
Teresa Charlin611c7fb2022-01-07 09:47:29 +0000478 case LayerType::Input :
479 {
480 auto inputQueueDescriptor = PolymorphicDowncast<const InputQueueDescriptor*>(&descriptor);
481 return std::make_unique<CopyMemGenericWorkload>(*inputQueueDescriptor, info);
482 }
483 case LayerType::InstanceNormalization :
484 {
485 auto instanceNormalizationQueueDescriptor
486 = PolymorphicDowncast<const InstanceNormalizationQueueDescriptor*>(&descriptor);
487 return MakeWorkload<ClInstanceNormalizationWorkload>(*instanceNormalizationQueueDescriptor,
488 info,
489 m_CLCompileContext);
490 }
491 case LayerType::L2Normalization :
492 {
493 auto l2NormalizationQueueDescriptor
494 = PolymorphicDowncast<const L2NormalizationQueueDescriptor*>(&descriptor);
495 return MakeWorkload<ClL2NormalizationFloatWorkload, NullWorkload>(*l2NormalizationQueueDescriptor,
496 info,
497 m_CLCompileContext);
498 }
499 case LayerType::LogicalBinary :
500 {
501 auto logicalBinaryQueueDescriptor = PolymorphicDowncast<const LogicalBinaryQueueDescriptor*>(&descriptor);
502
503 switch(logicalBinaryQueueDescriptor->m_Parameters.m_Operation)
504 {
505 case LogicalBinaryOperation::LogicalAnd:
506 return std::make_unique<ClLogicalAndWorkload>(*logicalBinaryQueueDescriptor,
507 info,
508 m_CLCompileContext);
509 case LogicalBinaryOperation::LogicalOr:
510 return std::make_unique<ClLogicalOrWorkload>(*logicalBinaryQueueDescriptor,
511 info,
512 m_CLCompileContext);
513 default:
514 return nullptr;
515 }
516 }
517 case LayerType::LogSoftmax :
518 {
519 auto logSoftmaxQueueDescriptor = PolymorphicDowncast<const LogSoftmaxQueueDescriptor*>(&descriptor);
520
521 return MakeWorkload<ClLogSoftmaxWorkload>(*logSoftmaxQueueDescriptor,
522 info,
523 m_MemoryManager->GetIntraLayerManager(),
524 m_CLCompileContext);
525 }
526 case LayerType::Lstm :
527 {
528 auto lstmQueueDescriptor = PolymorphicDowncast<const LstmQueueDescriptor*>(&descriptor);
529 return MakeWorkload<ClLstmFloatWorkload, NullWorkload>(*lstmQueueDescriptor, info, m_CLCompileContext);
530 }
531 case LayerType::Maximum :
532 {
533 auto maximumQueueDescriptor = PolymorphicDowncast<const MaximumQueueDescriptor*>(&descriptor);
534 return MakeWorkload<ClMaximumWorkload>(*maximumQueueDescriptor, info, m_CLCompileContext);
535 }
536 case LayerType::Mean :
537 {
538 auto meanQueueDescriptor = PolymorphicDowncast<const MeanQueueDescriptor*>(&descriptor);
539 return MakeWorkload<ClMeanWorkload>(*meanQueueDescriptor, info, m_CLCompileContext);
540 }
541 case LayerType::MemCopy :
542 {
543 auto memCopyQueueDescriptor = PolymorphicDowncast<const MemCopyQueueDescriptor*>(&descriptor);
544 if (memCopyQueueDescriptor->m_Inputs.empty() || !memCopyQueueDescriptor->m_Inputs[0])
545 {
546 throw InvalidArgumentException("ClWorkloadFactory: Invalid null input for MemCopy workload");
547 }
548 return MakeWorkload<CopyMemGenericWorkload>(*memCopyQueueDescriptor, info);
549 }
550 case LayerType::MemImport :
551 {
552 auto memImportQueueDescriptor = PolymorphicDowncast<const MemImportQueueDescriptor*>(&descriptor);
553 if (memImportQueueDescriptor->m_Inputs.empty() || !memImportQueueDescriptor->m_Inputs[0])
554 {
555 throw InvalidArgumentException("ClWorkloadFactory: Invalid null input for MemImport workload");
556 }
557 return std::make_unique<ImportMemGenericWorkload>(*memImportQueueDescriptor, info);
558 }
559 case LayerType::Minimum :
560 {
561 auto minimumQueueDescriptor = PolymorphicDowncast<const MinimumQueueDescriptor*>(&descriptor);
562 return MakeWorkload<ClMinimumWorkload>(*minimumQueueDescriptor, info, m_CLCompileContext);
563 }
564 case LayerType::Multiplication :
565 {
566 auto multiplicationQueueDescriptor = PolymorphicDowncast<const MultiplicationQueueDescriptor*>(&descriptor);
567 return MakeWorkload<ClMultiplicationWorkload>(*multiplicationQueueDescriptor, info, m_CLCompileContext);
568 }
569 case LayerType::Normalization :
570 {
571 auto normalizationQueueDescriptor = PolymorphicDowncast<const NormalizationQueueDescriptor*>(&descriptor);
572 return MakeWorkload<ClNormalizationFloatWorkload, NullWorkload>(*normalizationQueueDescriptor,
573 info,
574 m_CLCompileContext);
575 }
576 case LayerType::Output :
577 {
578 auto outputQueueDescriptor = PolymorphicDowncast<const OutputQueueDescriptor*>(&descriptor);
579 return std::make_unique<CopyMemGenericWorkload>(*outputQueueDescriptor, info);
580 }
581 case LayerType::Pad :
582 {
583 auto padQueueDescriptor = PolymorphicDowncast<const PadQueueDescriptor*>(&descriptor);
584 return MakeWorkload<ClPadWorkload>(*padQueueDescriptor, info, m_CLCompileContext);
585 }
586 case LayerType::Permute :
587 {
588 auto permuteQueueDescriptor = PolymorphicDowncast<const PermuteQueueDescriptor*>(&descriptor);
589 return MakeWorkload<ClPermuteWorkload>(*permuteQueueDescriptor, info, m_CLCompileContext);
590 }
591 case LayerType::Pooling2d :
592 {
593 auto pooling2dQueueDescriptor = PolymorphicDowncast<const Pooling2dQueueDescriptor*>(&descriptor);
594 return MakeWorkload<ClPooling2dWorkload>(*pooling2dQueueDescriptor, info, m_CLCompileContext);
595 }
Ryan OSheabab8fa92022-03-09 10:29:02 +0000596 case LayerType::Pooling3d :
597 {
598 auto pooling3dQueueDescriptor = PolymorphicDowncast<const Pooling3dQueueDescriptor*>(&descriptor);
599 return MakeWorkload<ClPooling3dWorkload>(*pooling3dQueueDescriptor, info, m_CLCompileContext);
600 }
Teresa Charlin611c7fb2022-01-07 09:47:29 +0000601 case LayerType::PreCompiled :
602 {
603 auto preCompiledQueueDescriptor = PolymorphicDowncast<const PreCompiledQueueDescriptor*>(&descriptor);
604 return MakeWorkload<NullWorkload, NullWorkload>(*preCompiledQueueDescriptor, info, m_CLCompileContext);
605 }
606 case LayerType::Prelu :
607 {
608 auto preluQueueDescriptor = PolymorphicDowncast<const PreluQueueDescriptor*>(&descriptor);
609 return MakeWorkload<ClPreluWorkload>(*preluQueueDescriptor, info, m_CLCompileContext);
610 }
611 case LayerType::QLstm :
612 {
613 auto qLstmQueueDescriptor = PolymorphicDowncast<const QLstmQueueDescriptor*>(&descriptor);
614 return std::make_unique<ClQLstmWorkload>(*qLstmQueueDescriptor, info, m_CLCompileContext);
615 }
616 case LayerType::Quantize :
617 {
618 auto quantizeQueueDescriptor = PolymorphicDowncast<const QuantizeQueueDescriptor*>(&descriptor);
619 return MakeWorkload<ClQuantizeWorkload>(*quantizeQueueDescriptor, info, m_CLCompileContext);
620 }
621 case LayerType::QuantizedLstm :
622 {
623 auto quantizedLstmQueueDescriptor = PolymorphicDowncast<const QuantizedLstmQueueDescriptor*>(&descriptor);
624 return MakeWorkload<ClQuantizedLstmWorkload>(*quantizedLstmQueueDescriptor, info, m_CLCompileContext);
625 }
626 case LayerType::Rank :
627 {
628 auto rankQueueDescriptor = PolymorphicDowncast<const RankQueueDescriptor*>(&descriptor);
629 return std::make_unique<ClRankWorkload>(*rankQueueDescriptor, info);
630 }
631 case LayerType::Reduce :
632 {
633 auto reduceQueueDescriptor = PolymorphicDowncast<const ReduceQueueDescriptor*>(&descriptor);
634 return std::make_unique<ClReduceWorkload>(*reduceQueueDescriptor, info);
635 }
636 case LayerType::Reshape :
637 {
638 auto reshapeQueueDescriptor = PolymorphicDowncast<const ReshapeQueueDescriptor*>(&descriptor);
639 return MakeWorkload<ClReshapeWorkload>(*reshapeQueueDescriptor, info, m_CLCompileContext);
640 }
641 case LayerType::Resize :
642 {
643 auto resizeQueueDescriptor = PolymorphicDowncast<const ResizeQueueDescriptor*>(&descriptor);
644 return MakeWorkload<ClResizeWorkload>(*resizeQueueDescriptor, info, m_CLCompileContext);
645 }
646 case LayerType::Slice :
647 {
648 auto sliceQueueDescriptor = PolymorphicDowncast<const SliceQueueDescriptor*>(&descriptor);
649 return MakeWorkload<ClSliceWorkload>(*sliceQueueDescriptor, info, m_CLCompileContext);
650 }
651 case LayerType::Softmax :
652 {
653 auto softmaxQueueDescriptor = PolymorphicDowncast<const SoftmaxQueueDescriptor*>(&descriptor);
654 return std::make_unique<ClSoftmaxWorkload>(*softmaxQueueDescriptor,
655 info,
656 m_MemoryManager->GetIntraLayerManager(),
657 m_CLCompileContext);
658 }
659 case LayerType::SpaceToBatchNd :
660 {
661 auto spaceToBatchNdQueueDescriptor
662 = PolymorphicDowncast<const SpaceToBatchNdQueueDescriptor*>(&descriptor);
663 return MakeWorkload<ClSpaceToBatchNdWorkload>(*spaceToBatchNdQueueDescriptor, info, m_CLCompileContext);
664 }
665 case LayerType::SpaceToDepth :
666 {
667 auto spaceToDepthQueueDescriptor = PolymorphicDowncast<const SpaceToDepthQueueDescriptor*>(&descriptor);
668 return MakeWorkload<ClSpaceToDepthWorkload>(*spaceToDepthQueueDescriptor, info, m_CLCompileContext);
669 }
670 case LayerType::Splitter :
671 {
672 auto splitterQueueDescriptor = PolymorphicDowncast<const SplitterQueueDescriptor*>(&descriptor);
673 return MakeWorkload<ClSplitterWorkload>(*splitterQueueDescriptor, info, m_CLCompileContext);
674 }
675 case LayerType::Stack :
676 {
677 auto stackQueueDescriptor = PolymorphicDowncast<const StackQueueDescriptor*>(&descriptor);
678 return MakeWorkload<ClStackWorkload>(*stackQueueDescriptor, info, m_CLCompileContext);
679 }
680 case LayerType::StridedSlice :
681 {
682 auto stridedSliceQueueDescriptor = PolymorphicDowncast<const StridedSliceQueueDescriptor*>(&descriptor);
683 return MakeWorkload<ClStridedSliceWorkload>(*stridedSliceQueueDescriptor, info, m_CLCompileContext);
684 }
685 case LayerType::Subtraction :
686 {
687 auto subtractionQueueDescriptor = PolymorphicDowncast<const SubtractionQueueDescriptor*>(&descriptor);
688 return MakeWorkload<ClSubtractionWorkload>(*subtractionQueueDescriptor, info, m_CLCompileContext);
689 }
690 case LayerType::Transpose :
691 {
692 auto transposeQueueDescriptor = PolymorphicDowncast<const TransposeQueueDescriptor*>(&descriptor);
693 return MakeWorkload<ClTransposeWorkload>(*transposeQueueDescriptor, info, m_CLCompileContext);
694 }
695 case LayerType::TransposeConvolution2d :
696 {
697 auto transposeConvolution2dQueueDescriptor
698 = PolymorphicDowncast<const TransposeConvolution2dQueueDescriptor*>(&descriptor);
699 return MakeWorkload<ClTransposeConvolution2dWorkload>(*transposeConvolution2dQueueDescriptor,
700 info,
701 m_MemoryManager->GetIntraLayerManager(),
702 m_CLCompileContext);
703 }
Cathal Corbett4952a3e2022-03-03 15:14:18 +0000704 case LayerType::UnidirectionalSequenceLstm :
705 {
706 auto desc = PolymorphicDowncast<const UnidirectionalSequenceLstmQueueDescriptor*>(&descriptor);
707 return MakeWorkloadHelper<ClUnidirectionalSequenceLstmFloatWorkload, NullWorkload>(*desc,
708 info,
709 m_CLCompileContext);
710 }
Teresa Charlin611c7fb2022-01-07 09:47:29 +0000711 default:
712 return nullptr;
713 }
714}
715
telsoa014fcda012018-03-09 14:13:49 +0000716std::unique_ptr<IWorkload> ClWorkloadFactory::CreateActivation(const ActivationQueueDescriptor& descriptor,
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100717 const WorkloadInfo& info) const
telsoa014fcda012018-03-09 14:13:49 +0000718{
Sadik Armagane9444752020-12-02 11:28:58 +0000719 return MakeWorkload<ClActivationWorkload>(descriptor, info, m_CLCompileContext);
telsoa014fcda012018-03-09 14:13:49 +0000720}
721
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100722std::unique_ptr<IWorkload> ClWorkloadFactory::CreateAddition(const AdditionQueueDescriptor& descriptor,
723 const WorkloadInfo& info) const
telsoa014fcda012018-03-09 14:13:49 +0000724{
Sadik Armagane9444752020-12-02 11:28:58 +0000725 return MakeWorkload<ClAdditionWorkload>(descriptor, info, m_CLCompileContext);
telsoa014fcda012018-03-09 14:13:49 +0000726}
727
James Conroy2dc05722019-09-19 17:00:31 +0100728std::unique_ptr<IWorkload> ClWorkloadFactory::CreateArgMinMax(const ArgMinMaxQueueDescriptor& descriptor,
729 const WorkloadInfo& info) const
730{
Sadik Armagane9444752020-12-02 11:28:58 +0000731 return std::make_unique<ClArgMinMaxWorkload>(descriptor, info, m_CLCompileContext);
James Conroy2dc05722019-09-19 17:00:31 +0100732}
733
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100734std::unique_ptr<IWorkload> ClWorkloadFactory::CreateBatchNormalization(
735 const BatchNormalizationQueueDescriptor& descriptor,
736 const WorkloadInfo& info) const
telsoa014fcda012018-03-09 14:13:49 +0000737{
Sadik Armagane9444752020-12-02 11:28:58 +0000738 return MakeWorkload<ClBatchNormalizationFloatWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
telsoa014fcda012018-03-09 14:13:49 +0000739}
740
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100741std::unique_ptr<IWorkload> ClWorkloadFactory::CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor& descriptor,
telsoa014fcda012018-03-09 14:13:49 +0000742 const WorkloadInfo& info) const
743{
Sadik Armagane9444752020-12-02 11:28:58 +0000744 return MakeWorkload<ClBatchToSpaceNdWorkload>(descriptor, info, m_CLCompileContext);
telsoa014fcda012018-03-09 14:13:49 +0000745}
746
Sadik Armaganf40d6d42021-04-22 09:12:11 +0100747std::unique_ptr<IWorkload> ClWorkloadFactory::CreateCast(const CastQueueDescriptor& descriptor,
748 const WorkloadInfo& info) const
749{
750 return MakeWorkload<ClCastWorkload>(descriptor, info, m_CLCompileContext);
751}
752
Teresa Charlin1222dbd2021-09-02 13:58:52 +0100753std::unique_ptr<IWorkload> ClWorkloadFactory::CreateChannelShuffle(const ChannelShuffleQueueDescriptor& descriptor,
754 const WorkloadInfo& info) const
755{
756 return MakeWorkload<ClChannelShuffleWorkload>(descriptor, info, m_CLCompileContext);
757}
758
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100759std::unique_ptr<IWorkload> ClWorkloadFactory::CreateComparison(const ComparisonQueueDescriptor& descriptor,
760 const WorkloadInfo& info) const
761{
Sadik Armagane9444752020-12-02 11:28:58 +0000762 return MakeWorkload<ClComparisonWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100763}
764
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100765std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConcat(const ConcatQueueDescriptor& descriptor,
766 const WorkloadInfo& info) const
Jim Flynn4ed6c832019-05-20 11:02:46 +0100767{
Sadik Armagane9444752020-12-02 11:28:58 +0000768 return MakeWorkload<ClConcatWorkload>(descriptor, info, m_CLCompileContext);
Jim Flynn4ed6c832019-05-20 11:02:46 +0100769}
770
telsoa014fcda012018-03-09 14:13:49 +0000771std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConstant(const ConstantQueueDescriptor& descriptor,
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100772 const WorkloadInfo& info) const
telsoa014fcda012018-03-09 14:13:49 +0000773{
Sadik Armagane9444752020-12-02 11:28:58 +0000774 return MakeWorkload<ClConstantWorkload>(descriptor, info, m_CLCompileContext);
telsoa014fcda012018-03-09 14:13:49 +0000775}
776
telsoa01c577f2c2018-08-31 09:22:23 +0100777std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConvertFp16ToFp32(
778 const ConvertFp16ToFp32QueueDescriptor& descriptor,
779 const WorkloadInfo& info) const
780{
Sadik Armagane9444752020-12-02 11:28:58 +0000781 return MakeWorkload<ClConvertFp16ToFp32Workload>(descriptor, info, m_CLCompileContext);
telsoa01c577f2c2018-08-31 09:22:23 +0100782}
783
784std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConvertFp32ToFp16(
785 const ConvertFp32ToFp16QueueDescriptor& descriptor,
786 const WorkloadInfo& info) const
787{
Sadik Armagane9444752020-12-02 11:28:58 +0000788 return MakeWorkload<ClConvertFp32ToFp16Workload>(descriptor, info, m_CLCompileContext);
telsoa01c577f2c2018-08-31 09:22:23 +0100789}
790
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100791std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConvolution2d(const Convolution2dQueueDescriptor& descriptor,
792 const WorkloadInfo& info) const
793{
Sadik Armagan04a72972020-09-14 15:44:18 +0100794 bool isFastMathEnabled = false;
795 if (m_ModelContextPtr)
796 {
797 if (m_ModelContextPtr.get() != nullptr)
798 {
799 auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
800 if (modelOptions)
801 {
802 isFastMathEnabled = modelOptions->IsFastMathEnabled();
803 }
804 }
805 }
806 return MakeWorkload<ClConvolution2dWorkload>(descriptor,
807 info,
808 m_MemoryManager->GetIntraLayerManager(),
Sadik Armagane9444752020-12-02 11:28:58 +0000809 m_CLCompileContext,
Sadik Armagan04a72972020-09-14 15:44:18 +0100810 isFastMathEnabled);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100811}
812
Teresa Charlin615ad6c2021-10-26 12:22:20 +0100813std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConvolution3d(const Convolution3dQueueDescriptor& descriptor,
814 const WorkloadInfo& info) const
815{
816 bool isFastMathEnabled = false;
817 if (m_ModelContextPtr)
818 {
819 if (m_ModelContextPtr.get() != nullptr)
820 {
821 auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
822 if (modelOptions)
823 {
824 isFastMathEnabled = modelOptions->IsFastMathEnabled();
825 }
826 }
827 }
828 return MakeWorkload<ClConvolution3dWorkload>(descriptor,
829 info,
830 m_MemoryManager->GetIntraLayerManager(),
831 m_CLCompileContext,
832 isFastMathEnabled);
833}
834
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100835std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDebug(const DebugQueueDescriptor& descriptor,
836 const WorkloadInfo& info) const
837{
Sadik Armagane9444752020-12-02 11:28:58 +0000838 return MakeWorkload<NullWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100839}
840
841std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDepthToSpace(const DepthToSpaceQueueDescriptor& descriptor,
842 const WorkloadInfo& info) const
843{
Sadik Armagane9444752020-12-02 11:28:58 +0000844 return MakeWorkload<ClDepthToSpaceWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100845}
846
847std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDepthwiseConvolution2d(
848 const DepthwiseConvolution2dQueueDescriptor& descriptor,
849 const WorkloadInfo& info) const
850{
Sadik Armagane9444752020-12-02 11:28:58 +0000851 return MakeWorkload<ClDepthwiseConvolutionWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100852}
853
854std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDequantize(const DequantizeQueueDescriptor& descriptor,
855 const WorkloadInfo& info) const
856{
Sadik Armagane9444752020-12-02 11:28:58 +0000857 return MakeWorkload<ClDequantizeWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100858}
859
860std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDetectionPostProcess(
861 const DetectionPostProcessQueueDescriptor& descriptor,
862 const WorkloadInfo& info) const
863{
Sadik Armagane9444752020-12-02 11:28:58 +0000864 return MakeWorkload<NullWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100865}
866
867std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDivision(const DivisionQueueDescriptor& descriptor,
868 const WorkloadInfo& info) const
869{
Teresa Charline11e63d2021-04-21 12:56:45 +0100870 return std::make_unique<ClDivisionWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100871}
872
josh minor4a3c6102020-01-06 16:40:46 -0600873std::unique_ptr<IWorkload> ClWorkloadFactory::CreateElementwiseUnary(const ElementwiseUnaryQueueDescriptor& descriptor,
874 const WorkloadInfo& info) const
875{
Sadik Armagan9fabf432020-05-27 13:40:58 +0100876 switch(descriptor.m_Parameters.m_Operation)
josh minor4a3c6102020-01-06 16:40:46 -0600877 {
Sadik Armagan9fabf432020-05-27 13:40:58 +0100878 case UnaryOperation::Abs:
James Conroyfe3ec942020-11-18 14:20:53 +0000879 {
880 AbsQueueDescriptor absQueueDescriptor;
881 absQueueDescriptor.m_Inputs = descriptor.m_Inputs;
882 absQueueDescriptor.m_Outputs = descriptor.m_Outputs;
josh minor4a3c6102020-01-06 16:40:46 -0600883
Sadik Armagane9444752020-12-02 11:28:58 +0000884 return std::make_unique<ClAbsWorkload>(absQueueDescriptor, info, m_CLCompileContext);
James Conroyfe3ec942020-11-18 14:20:53 +0000885 }
Sadik Armagan9fabf432020-05-27 13:40:58 +0100886 case UnaryOperation::Exp:
Sadik Armagane9444752020-12-02 11:28:58 +0000887 return std::make_unique<ClExpWorkload>(descriptor, info, m_CLCompileContext);
Teresa Charlin50de4fa2021-05-31 18:47:33 +0100888 case UnaryOperation::Log:
889 return std::make_unique<ClLogWorkload>(descriptor, info, m_CLCompileContext);
890 case UnaryOperation::LogicalNot:
891 return std::make_unique<ClLogicalNotWorkload>(descriptor, info, m_CLCompileContext);
Sadik Armagan9fabf432020-05-27 13:40:58 +0100892 case UnaryOperation::Neg:
Sadik Armagane9444752020-12-02 11:28:58 +0000893 return std::make_unique<ClNegWorkload>(descriptor, info, m_CLCompileContext);
Sadik Armagan9fabf432020-05-27 13:40:58 +0100894 case UnaryOperation::Rsqrt:
James Conroyfe3ec942020-11-18 14:20:53 +0000895 {
896 RsqrtQueueDescriptor rsqrtQueueDescriptor;
897 rsqrtQueueDescriptor.m_Inputs = descriptor.m_Inputs;
898 rsqrtQueueDescriptor.m_Outputs = descriptor.m_Outputs;
josh minor4a3c6102020-01-06 16:40:46 -0600899
Sadik Armagane9444752020-12-02 11:28:58 +0000900 return std::make_unique<ClRsqrtWorkload>(rsqrtQueueDescriptor, info, m_CLCompileContext);
James Conroyfe3ec942020-11-18 14:20:53 +0000901 }
Teresa Charlin50de4fa2021-05-31 18:47:33 +0100902 case UnaryOperation::Sin:
903 return std::make_unique<ClSinWorkload>(descriptor, info, m_CLCompileContext);
Sadik Armagan9fabf432020-05-27 13:40:58 +0100904 default:
905 return nullptr;
josh minor4a3c6102020-01-06 16:40:46 -0600906 }
josh minor4a3c6102020-01-06 16:40:46 -0600907}
908
Sadik Armagan66aecb02020-06-24 11:42:20 +0100909std::unique_ptr<IWorkload> ClWorkloadFactory::CreateFill(const FillQueueDescriptor& descriptor,
910 const WorkloadInfo& info) const
911{
Sadik Armagane9444752020-12-02 11:28:58 +0000912 return std::make_unique<ClFillWorkload>(descriptor, info, m_CLCompileContext);
Sadik Armagan66aecb02020-06-24 11:42:20 +0100913}
914
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100915std::unique_ptr<IWorkload> ClWorkloadFactory::CreateFloor(const FloorQueueDescriptor& descriptor,
916 const WorkloadInfo& info) const
917{
Sadik Armagane9444752020-12-02 11:28:58 +0000918 return MakeWorkload<ClFloorFloatWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100919}
920
921std::unique_ptr<IWorkload> ClWorkloadFactory::CreateFullyConnected(const FullyConnectedQueueDescriptor& descriptor,
922 const WorkloadInfo& info) const
923{
Sadik Armagane9444752020-12-02 11:28:58 +0000924 return MakeWorkload<ClFullyConnectedWorkload>(descriptor,
925 info,
926 m_MemoryManager->GetIntraLayerManager(),
927 m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100928}
929
930std::unique_ptr<IWorkload> ClWorkloadFactory::CreateGather(const GatherQueueDescriptor& descriptor,
931 const WorkloadInfo& info) const
932{
Sadik Armagane9444752020-12-02 11:28:58 +0000933 return MakeWorkload<ClGatherWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100934}
935
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100936std::unique_ptr<IWorkload> ClWorkloadFactory::CreateInput(const InputQueueDescriptor& descriptor,
937 const WorkloadInfo& info) const
938{
939 return std::make_unique<CopyMemGenericWorkload>(descriptor, info);
940}
941
942std::unique_ptr<IWorkload> ClWorkloadFactory::CreateInstanceNormalization(
943 const InstanceNormalizationQueueDescriptor& descriptor,
944 const WorkloadInfo& info) const
945{
Sadik Armagane9444752020-12-02 11:28:58 +0000946 return MakeWorkload<ClInstanceNormalizationWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100947}
948
949std::unique_ptr<IWorkload> ClWorkloadFactory::CreateL2Normalization(const L2NormalizationQueueDescriptor& descriptor,
950 const WorkloadInfo& info) const
951{
Sadik Armagane9444752020-12-02 11:28:58 +0000952 return MakeWorkload<ClL2NormalizationFloatWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100953}
954
James Conroyfe3ec942020-11-18 14:20:53 +0000955std::unique_ptr<IWorkload> ClWorkloadFactory::CreateLogicalBinary(const LogicalBinaryQueueDescriptor& descriptor,
956 const WorkloadInfo& info) const
957{
958 switch(descriptor.m_Parameters.m_Operation)
959 {
960 case LogicalBinaryOperation::LogicalAnd:
Sadik Armagane9444752020-12-02 11:28:58 +0000961 return std::make_unique<ClLogicalAndWorkload>(descriptor, info, m_CLCompileContext);
James Conroyfe3ec942020-11-18 14:20:53 +0000962 case LogicalBinaryOperation::LogicalOr:
Sadik Armagane9444752020-12-02 11:28:58 +0000963 return std::make_unique<ClLogicalOrWorkload>(descriptor, info, m_CLCompileContext);
James Conroyfe3ec942020-11-18 14:20:53 +0000964 default:
965 return nullptr;
966 }
967}
968
Teresa Charlin8398edc2020-07-20 14:23:02 +0100969std::unique_ptr<IWorkload> ClWorkloadFactory::CreateLogSoftmax(const LogSoftmaxQueueDescriptor& descriptor,
970 const WorkloadInfo& info) const
971{
Sadik Armagane9444752020-12-02 11:28:58 +0000972 return MakeWorkload<ClLogSoftmaxWorkload>(descriptor,
973 info,
974 m_MemoryManager->GetIntraLayerManager(),
975 m_CLCompileContext);
Teresa Charlin8398edc2020-07-20 14:23:02 +0100976}
977
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100978std::unique_ptr<IWorkload> ClWorkloadFactory::CreateLstm(const LstmQueueDescriptor& descriptor,
979 const WorkloadInfo& info) const
980{
Sadik Armagane9444752020-12-02 11:28:58 +0000981 return MakeWorkload<ClLstmFloatWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100982}
983
Nattapat Chaimanowong5a4304a2018-11-28 10:44:37 +0000984std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMaximum(const MaximumQueueDescriptor& descriptor,
985 const WorkloadInfo& info) const
986{
Sadik Armagane9444752020-12-02 11:28:58 +0000987 return MakeWorkload<ClMaximumWorkload>(descriptor, info, m_CLCompileContext);
Nattapat Chaimanowong5a4304a2018-11-28 10:44:37 +0000988}
989
narpra01a6bf9122018-09-10 09:50:09 +0100990std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMean(const MeanQueueDescriptor& descriptor,
991 const WorkloadInfo& info) const
992{
Sadik Armagane9444752020-12-02 11:28:58 +0000993 return MakeWorkload<ClMeanWorkload>(descriptor, info, m_CLCompileContext);
narpra01a6bf9122018-09-10 09:50:09 +0100994}
995
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100996std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMemCopy(const MemCopyQueueDescriptor& descriptor,
997 const WorkloadInfo& info) const
jimfly012c9322a2018-09-19 10:59:49 +0100998{
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100999 if (descriptor.m_Inputs.empty() || !descriptor.m_Inputs[0])
1000 {
1001 throw InvalidArgumentException("ClWorkloadFactory: Invalid null input for MemCopy workload");
1002 }
1003
1004 return MakeWorkload<CopyMemGenericWorkload>(descriptor, info);
jimfly012c9322a2018-09-19 10:59:49 +01001005}
1006
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001007std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMemImport(const MemImportQueueDescriptor& descriptor,
1008 const WorkloadInfo& info) const
FrancisMurtagh20995952018-12-17 12:11:36 +00001009{
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001010 if (descriptor.m_Inputs.empty() || !descriptor.m_Inputs[0])
1011 {
1012 throw InvalidArgumentException("ClWorkloadFactory: Invalid null input for MemImport workload");
1013 }
1014
1015 return std::make_unique<ImportMemGenericWorkload>(descriptor, info);
FrancisMurtagh20995952018-12-17 12:11:36 +00001016}
1017
kevmay0190539692018-11-29 08:40:19 +00001018std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMinimum(const MinimumQueueDescriptor& descriptor,
1019 const WorkloadInfo& info) const
1020{
Sadik Armagane9444752020-12-02 11:28:58 +00001021 return MakeWorkload<ClMinimumWorkload>(descriptor, info, m_CLCompileContext);
kevmay0190539692018-11-29 08:40:19 +00001022}
1023
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001024std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMultiplication(const MultiplicationQueueDescriptor& descriptor,
1025 const WorkloadInfo& info) const
1026{
Sadik Armagane9444752020-12-02 11:28:58 +00001027 return MakeWorkload<ClMultiplicationWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001028}
1029
1030std::unique_ptr<IWorkload> ClWorkloadFactory::CreateNormalization(const NormalizationQueueDescriptor& descriptor,
1031 const WorkloadInfo& info) const
1032{
Sadik Armagane9444752020-12-02 11:28:58 +00001033 return MakeWorkload<ClNormalizationFloatWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001034}
1035
1036std::unique_ptr<IWorkload> ClWorkloadFactory::CreateOutput(const OutputQueueDescriptor& descriptor,
1037 const WorkloadInfo& info) const
1038{
1039 return std::make_unique<CopyMemGenericWorkload>(descriptor, info);
1040}
1041
1042std::unique_ptr<IWorkload> ClWorkloadFactory::CreatePad(const PadQueueDescriptor& descriptor,
1043 const WorkloadInfo& info) const
1044{
Sadik Armagane9444752020-12-02 11:28:58 +00001045 return MakeWorkload<ClPadWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001046}
1047
1048std::unique_ptr<IWorkload> ClWorkloadFactory::CreatePermute(const PermuteQueueDescriptor& descriptor,
Matteo Martincigh59a950c2018-12-13 12:48:25 +00001049 const WorkloadInfo& info) const
1050{
Sadik Armagane9444752020-12-02 11:28:58 +00001051 return MakeWorkload<ClPermuteWorkload>(descriptor, info, m_CLCompileContext);
Matteo Martincigh59a950c2018-12-13 12:48:25 +00001052}
1053
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001054std::unique_ptr<IWorkload> ClWorkloadFactory::CreatePooling2d(const Pooling2dQueueDescriptor& descriptor,
1055 const WorkloadInfo& info) const
Nattapat Chaimanowonga9a1cf12018-12-03 16:06:49 +00001056{
Sadik Armagane9444752020-12-02 11:28:58 +00001057 return MakeWorkload<ClPooling2dWorkload>(descriptor, info, m_CLCompileContext);
Mohamed Nour Abouelseouda1d3c6a2018-12-27 12:39:16 +00001058}
1059
Matteo Martincigh49124022019-01-11 13:25:59 +00001060std::unique_ptr<IWorkload> ClWorkloadFactory::CreatePreCompiled(const PreCompiledQueueDescriptor& descriptor,
1061 const WorkloadInfo& info) const
1062{
Sadik Armagane9444752020-12-02 11:28:58 +00001063 return MakeWorkload<NullWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
Matteo Martincigh49124022019-01-11 13:25:59 +00001064}
1065
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001066std::unique_ptr<IWorkload> ClWorkloadFactory::CreatePrelu(const PreluQueueDescriptor &descriptor,
1067 const WorkloadInfo &info) const
narpra01b89b05f2019-01-16 09:53:09 +00001068{
Sadik Armagane9444752020-12-02 11:28:58 +00001069 return MakeWorkload<ClPreluWorkload>(descriptor, info, m_CLCompileContext);
narpra01b89b05f2019-01-16 09:53:09 +00001070}
1071
Ryan OShea2323af42020-05-13 16:36:19 +01001072std::unique_ptr<IWorkload> ClWorkloadFactory::CreateQLstm(const QLstmQueueDescriptor& descriptor,
1073 const WorkloadInfo& info) const
1074{
Sadik Armagane9444752020-12-02 11:28:58 +00001075 return std::make_unique<ClQLstmWorkload>(descriptor, info, m_CLCompileContext);
Ryan OShea2323af42020-05-13 16:36:19 +01001076}
1077
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001078std::unique_ptr<IWorkload> ClWorkloadFactory::CreateQuantize(const QuantizeQueueDescriptor& descriptor,
1079 const WorkloadInfo& info) const
Aron Virginas-Tar7a3e2fe2019-06-27 18:54:47 +01001080{
Sadik Armagane9444752020-12-02 11:28:58 +00001081 return MakeWorkload<ClQuantizeWorkload>(descriptor, info, m_CLCompileContext);
James Conroyd2aa85e2019-07-01 17:12:40 +01001082}
1083
Ferran Balaguer737d9ff2019-08-01 09:58:08 +01001084std::unique_ptr<IWorkload> ClWorkloadFactory::CreateQuantizedLstm(const QuantizedLstmQueueDescriptor& descriptor,
1085 const WorkloadInfo& info) const
1086{
Sadik Armagane9444752020-12-02 11:28:58 +00001087 return MakeWorkload<ClQuantizedLstmWorkload>(descriptor, info, m_CLCompileContext);
Ferran Balaguer737d9ff2019-08-01 09:58:08 +01001088}
1089
David Monahanc11ba462020-12-03 11:09:46 +00001090std::unique_ptr<IWorkload> ClWorkloadFactory::CreateRank(const RankQueueDescriptor& descriptor,
1091 const WorkloadInfo& info) const
1092{
1093 return std::make_unique<ClRankWorkload>(descriptor, info);
1094}
1095
Sadik Armagana2747482021-02-09 10:28:54 +00001096std::unique_ptr<IWorkload> ClWorkloadFactory::CreateReduce(const ReduceQueueDescriptor& descriptor,
1097 const WorkloadInfo& info) const
1098{
1099 return std::make_unique<ClReduceWorkload>(descriptor, info);
1100}
1101
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001102std::unique_ptr<IWorkload> ClWorkloadFactory::CreateReshape(const ReshapeQueueDescriptor& descriptor,
1103 const WorkloadInfo& info) const
1104{
Sadik Armagane9444752020-12-02 11:28:58 +00001105 return MakeWorkload<ClReshapeWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001106}
1107
1108std::unique_ptr<IWorkload> ClWorkloadFactory::CreateResize(const ResizeQueueDescriptor& descriptor,
1109 const WorkloadInfo& info) const
1110{
Sadik Armagane9444752020-12-02 11:28:58 +00001111 return MakeWorkload<ClResizeWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001112}
1113
Aron Virginas-Tar94c4fef2019-11-25 15:37:08 +00001114std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSlice(const SliceQueueDescriptor& descriptor,
1115 const WorkloadInfo& info) const
1116{
Sadik Armagane9444752020-12-02 11:28:58 +00001117 return MakeWorkload<ClSliceWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar94c4fef2019-11-25 15:37:08 +00001118}
1119
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001120std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& descriptor,
1121 const WorkloadInfo& info) const
1122{
Sadik Armagane9444752020-12-02 11:28:58 +00001123 return std::make_unique<ClSoftmaxWorkload>(descriptor,
1124 info,
1125 m_MemoryManager->GetIntraLayerManager(),
1126 m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001127}
1128
1129std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSpaceToBatchNd(const SpaceToBatchNdQueueDescriptor& descriptor,
1130 const WorkloadInfo& info) const
1131{
Sadik Armagane9444752020-12-02 11:28:58 +00001132 return MakeWorkload<ClSpaceToBatchNdWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001133}
1134
1135std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSpaceToDepth(const SpaceToDepthQueueDescriptor& descriptor,
1136 const WorkloadInfo& info) const
1137{
Sadik Armagane9444752020-12-02 11:28:58 +00001138 return MakeWorkload<ClSpaceToDepthWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001139}
1140
1141std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSplitter(const SplitterQueueDescriptor& descriptor,
1142 const WorkloadInfo& info) const
1143{
Sadik Armagane9444752020-12-02 11:28:58 +00001144 return MakeWorkload<ClSplitterWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001145}
1146
Matthew Jacksond5166102019-07-31 14:06:28 +01001147std::unique_ptr<IWorkload> ClWorkloadFactory::CreateStack(const StackQueueDescriptor& descriptor,
1148 const WorkloadInfo& info) const
1149{
Sadik Armagane9444752020-12-02 11:28:58 +00001150 return MakeWorkload<ClStackWorkload>(descriptor, info, m_CLCompileContext);
Matthew Jacksond5166102019-07-31 14:06:28 +01001151}
1152
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001153std::unique_ptr<IWorkload> ClWorkloadFactory::CreateStridedSlice(const StridedSliceQueueDescriptor& descriptor,
Aron Virginas-Tarb2801962019-09-30 11:24:53 +01001154 const WorkloadInfo& info) const
1155{
Sadik Armagane9444752020-12-02 11:28:58 +00001156 return MakeWorkload<ClStridedSliceWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001157}
1158
1159std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSubtraction(const SubtractionQueueDescriptor& descriptor,
1160 const WorkloadInfo& info) const
1161{
Sadik Armagane9444752020-12-02 11:28:58 +00001162 return MakeWorkload<ClSubtractionWorkload>(descriptor, info, m_CLCompileContext);
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001163}
1164
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001165std::unique_ptr<IWorkload> ClWorkloadFactory::CreateTranspose(const TransposeQueueDescriptor& descriptor,
1166 const WorkloadInfo& info) const
1167{
Sadik Armagane9444752020-12-02 11:28:58 +00001168 return MakeWorkload<ClTransposeWorkload>(descriptor, info, m_CLCompileContext);
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001169}
1170
Aron Virginas-Tar8168f402019-10-04 13:10:16 +01001171std::unique_ptr<IWorkload> ClWorkloadFactory::CreateTransposeConvolution2d(
1172 const TransposeConvolution2dQueueDescriptor& descriptor,
1173 const WorkloadInfo& info) const
1174{
Sadik Armagane9444752020-12-02 11:28:58 +00001175 return MakeWorkload<ClTransposeConvolution2dWorkload>(descriptor,
1176 info,
1177 m_MemoryManager->GetIntraLayerManager(),
1178 m_CLCompileContext);
Aron Virginas-Tarb2801962019-09-30 11:24:53 +01001179}
1180
telsoa014fcda012018-03-09 14:13:49 +00001181} // namespace armnn