ArmNN
 22.02
ClWorkloadFactory.cpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #include "ClWorkloadFactory.hpp"
6 #include "ClBackendId.hpp"
10 
11 #include <Layer.hpp>
12 
13 #include <armnn/Exceptions.hpp>
14 #include <armnn/Logging.hpp>
15 #include <armnn/Utils.hpp>
19 
24 
25 #include <cl/ClTensorHandle.hpp>
28 
29 #include <arm_compute/core/CL/CLKernelLibrary.h>
30 #include <arm_compute/runtime/CL/CLBufferAllocator.h>
31 #include <arm_compute/runtime/CL/CLScheduler.h>
32 
34 #include <fstream>
35 
36 #include <sys/stat.h>
37 
38 namespace armnn
39 {
40 
41 namespace
42 {
43 static const BackendId s_Id{ClBackendId()};
44 }
45 
47  Optional<DataType> dataType,
48  std::string& outReasonIfUnsupported)
49 {
50  return IWorkloadFactory::IsLayerSupported(s_Id, layer, dataType, outReasonIfUnsupported);
51 }
52 
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 
62 {
63  return s_Id;
64 }
65 
67 {
68  if(m_ModelContextPtr)
69  {
70  auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
71  if (modelOptions->SaveCachedNetwork())
72  {
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));
87  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  }
92  }
93  }
94 
95  // 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  {
99  // Serialize ClContext to the file specified
100  std::ofstream file(filePath, std::ios::out | std::ios::binary);
101  serializer.SaveSerializedToStream(file);
102  }
103  }
104  }
105 }
106 
107 template <typename FloatWorkload, typename Uint8Workload, typename QueueDescriptorType, typename... Args>
108 std::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 
122 template <typename Workload, typename QueueDescriptorType, typename... Args>
123 std::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 
137 void ClWorkloadFactory::InitializeCLCompileContext()
138 {
139  // Initialize our m_CLCompileContext using default device and context
140  auto context = arm_compute::CLKernelLibrary::get().context();
141  auto device = arm_compute::CLKernelLibrary::get().get_device();
142  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();
149  if (!(modelOptions->SaveCachedNetwork()))
150  {
151  ClContextDeserializer deserializer;
152  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));
165  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  }
173  }
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  }
184  }
185  }
186 }
187 
188 ClWorkloadFactory::ClWorkloadFactory(const std::shared_ptr<ClMemoryManager>& memoryManager)
189  : m_MemoryManager(memoryManager), m_ModelContextPtr(IBackendInternal::IBackendSpecificModelContextPtr{})
190 {
191  InitializeCLCompileContext();
192 }
193 
194 ClWorkloadFactory::ClWorkloadFactory(const std::shared_ptr<ClMemoryManager>& memoryManager,
196  : m_MemoryManager(memoryManager), m_ModelContextPtr(modelContextPtr)
197 {
198  InitializeCLCompileContext();
199 }
200 
201 std::unique_ptr<ITensorHandle> ClWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo,
202  const bool IsMemoryManaged) const
203 {
204  IgnoreUnused(IsMemoryManaged);
205  std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo);
206  tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());
207 
208  return tensorHandle;
209 }
210 
211 std::unique_ptr<ITensorHandle> ClWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo,
212  DataLayout dataLayout,
213  const bool IsMemoryManaged) const
214 {
215  IgnoreUnused(IsMemoryManaged);
216  std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo, dataLayout);
217  tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());
218 
219  return tensorHandle;
220 }
221 
222 std::unique_ptr<ITensorHandle> ClWorkloadFactory::CreateSubTensorHandle(ITensorHandle& parent,
223  TensorShape const& subTensorShape,
224  unsigned int const* subTensorOrigin) const
225 {
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  {
232  // Arm compute indexes tensor coords in reverse order.
233  unsigned int revertedIndex = subTensorShape.GetNumDimensions() - i - 1;
234  coords.set(i, armnn::numeric_cast<int>(subTensorOrigin[revertedIndex]));
235  }
236 
237  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 
243  return std::make_unique<ClSubTensorHandle>(
244  PolymorphicDowncast<IClTensorHandle*>(&parent), shape, coords);
245 }
246 
247 std::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);
266  return std::make_unique<ClArgMinMaxWorkload>(*argMinMaxQueueDescriptor, info, m_CLCompileContext);
267  }
269  {
270  auto batchNormalizationQueueDescriptor
271  = PolymorphicDowncast<const BatchNormalizationQueueDescriptor*>(&descriptor);
272  return MakeWorkload<ClBatchNormalizationFloatWorkload, NullWorkload>
273  (*batchNormalizationQueueDescriptor, info, m_CLCompileContext);
274  }
276  {
277  auto batchToSpaceNdQueueDescriptor
278  = PolymorphicDowncast<const BatchToSpaceNdQueueDescriptor*>(&descriptor);
279  return MakeWorkload<ClBatchToSpaceNdWorkload>(*batchToSpaceNdQueueDescriptor, info, m_CLCompileContext);
280  }
281  case LayerType::Cast :
282  {
283  auto castQueueDescriptor = PolymorphicDowncast<const CastQueueDescriptor*>(&descriptor);
284  return MakeWorkload<ClCastWorkload>(*castQueueDescriptor, info, m_CLCompileContext);
285  }
287  {
288  auto channelShuffleQueueDescriptor
289  = PolymorphicDowncast<const ChannelShuffleQueueDescriptor*>(&descriptor);
290  return MakeWorkload<ClChannelShuffleWorkload>(*channelShuffleQueueDescriptor, info, m_CLCompileContext);
291  }
292  case LayerType::Comparison :
293  {
294  auto comparisonQueueDescriptor = PolymorphicDowncast<const ComparisonQueueDescriptor*>(&descriptor);
295  return MakeWorkload<ClComparisonWorkload>(*comparisonQueueDescriptor, info, m_CLCompileContext);
296  }
297  case LayerType::Concat :
298  {
299  auto concatQueueDescriptor = PolymorphicDowncast<const ConcatQueueDescriptor*>(&descriptor);
300  return MakeWorkload<ClConcatWorkload>(*concatQueueDescriptor, info, m_CLCompileContext);
301  }
302  case LayerType::Constant :
303  {
304  auto constantQueueDescriptor = PolymorphicDowncast<const ConstantQueueDescriptor*>(&descriptor);
305  return MakeWorkload<ClConstantWorkload>(*constantQueueDescriptor, info, m_CLCompileContext);
306  }
308  {
309  auto convertFp16ToFp32QueueDescriptor
310  = PolymorphicDowncast<const ConvertFp16ToFp32QueueDescriptor*>(&descriptor);
311  return MakeWorkload<ClConvertFp16ToFp32Workload>(*convertFp16ToFp32QueueDescriptor,
312  info,
313  m_CLCompileContext);
314  }
316  {
317  auto convertFp32ToFp16QueueDescriptor
318  = PolymorphicDowncast<const ConvertFp32ToFp16QueueDescriptor*>(&descriptor);
319  return MakeWorkload<ClConvertFp32ToFp16Workload>(*convertFp32ToFp16QueueDescriptor,
320  info,
321  m_CLCompileContext);
322  }
324  {
325  auto convolution2dQueueDescriptor = PolymorphicDowncast<const Convolution2dQueueDescriptor*>(&descriptor);
326 
327  bool isFastMathEnabled = false;
328  if (m_ModelContextPtr)
329  {
330  if (m_ModelContextPtr.get() != nullptr)
331  {
332  auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
333  if (modelOptions)
334  {
335  isFastMathEnabled = modelOptions->IsFastMathEnabled();
336  }
337  }
338  }
339  return MakeWorkload<ClConvolution2dWorkload>(*convolution2dQueueDescriptor,
340  info,
341  m_MemoryManager->GetIntraLayerManager(),
342  m_CLCompileContext,
343  isFastMathEnabled);
344  }
346  {
347  auto convolution3dQueueDescriptor = PolymorphicDowncast<const Convolution3dQueueDescriptor*>(&descriptor);
348 
349  bool isFastMathEnabled = false;
350  if (m_ModelContextPtr)
351  {
352  if (m_ModelContextPtr.get() != nullptr)
353  {
354  auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
355  if (modelOptions)
356  {
357  isFastMathEnabled = modelOptions->IsFastMathEnabled();
358  }
359  }
360  }
361  return MakeWorkload<ClConvolution3dWorkload>(*convolution3dQueueDescriptor,
362  info,
363  m_MemoryManager->GetIntraLayerManager(),
364  m_CLCompileContext,
365  isFastMathEnabled);
366  }
367  case LayerType::Debug :
368  {
369  auto debugQueueDescriptor = PolymorphicDowncast<const DebugQueueDescriptor*>(&descriptor);
370  return MakeWorkload<NullWorkload, NullWorkload>(*debugQueueDescriptor, info, m_CLCompileContext);
371  }
373  {
374  auto depthToSpaceQueueDescriptor = PolymorphicDowncast<const DepthToSpaceQueueDescriptor*>(&descriptor);
375  return MakeWorkload<ClDepthToSpaceWorkload>(*depthToSpaceQueueDescriptor, info, m_CLCompileContext);
376  }
378  {
379  auto depthwiseConvolution2dQueueDescriptor
380  = PolymorphicDowncast<const DepthwiseConvolution2dQueueDescriptor*>(&descriptor);
381  return MakeWorkload<ClDepthwiseConvolutionWorkload>(*depthwiseConvolution2dQueueDescriptor,
382  info,
383  m_CLCompileContext);
384  }
385  case LayerType::Dequantize :
386  {
387  auto dequantizeQueueDescriptor = PolymorphicDowncast<const DequantizeQueueDescriptor*>(&descriptor);
388  return MakeWorkload<ClDequantizeWorkload>(*dequantizeQueueDescriptor, info, m_CLCompileContext);
389  }
391  {
392  auto detectionPostProcessQueueDescriptor
393  = PolymorphicDowncast<const DetectionPostProcessQueueDescriptor*>(&descriptor);
394  return MakeWorkload<NullWorkload, NullWorkload>(*detectionPostProcessQueueDescriptor,
395  info,
396  m_CLCompileContext);
397  }
398  case LayerType::Division :
399  {
400  auto divisionQueueDescriptor = PolymorphicDowncast<const DivisionQueueDescriptor*>(&descriptor);
401  return std::make_unique<ClDivisionWorkload>(*divisionQueueDescriptor, info, m_CLCompileContext);
402  }
404  {
405  auto elementwiseUnaryQueueDescriptor
406  = PolymorphicDowncast<const ElementwiseUnaryQueueDescriptor*>(&descriptor);
407 
408  switch(elementwiseUnaryQueueDescriptor->m_Parameters.m_Operation)
409  {
410  case UnaryOperation::Abs:
411  {
412  AbsQueueDescriptor absQueueDescriptor;
413  absQueueDescriptor.m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs;
414  absQueueDescriptor.m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs;
415 
416  return std::make_unique<ClAbsWorkload>(absQueueDescriptor, info, m_CLCompileContext);
417  }
418  case UnaryOperation::Exp:
419  return std::make_unique<ClExpWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext);
420  case UnaryOperation::Log:
421  return std::make_unique<ClLogWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext);
423  return std::make_unique<ClLogicalNotWorkload>(*elementwiseUnaryQueueDescriptor,
424  info,
425  m_CLCompileContext);
426  case UnaryOperation::Neg:
427  return std::make_unique<ClNegWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext);
429  {
430  RsqrtQueueDescriptor rsqrtQueueDescriptor;
431  rsqrtQueueDescriptor.m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs;
432  rsqrtQueueDescriptor.m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs;
433 
434  return std::make_unique<ClRsqrtWorkload>(rsqrtQueueDescriptor, info, m_CLCompileContext);
435  }
436  case UnaryOperation::Sin:
437  return std::make_unique<ClSinWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext);
438  default:
439  return nullptr;
440  }
441  }
442  case LayerType::Fill :
443  {
444  auto fillQueueDescriptor = PolymorphicDowncast<const FillQueueDescriptor*>(&descriptor);
445  return std::make_unique<ClFillWorkload>(*fillQueueDescriptor, info, m_CLCompileContext);
446  }
447  case LayerType::Floor :
448  {
449  auto floorQueueDescriptor = PolymorphicDowncast<const FloorQueueDescriptor*>(&descriptor);
450  return MakeWorkload<ClFloorFloatWorkload, NullWorkload>(*floorQueueDescriptor, info, m_CLCompileContext);
451  }
453  {
454  auto fullyConnectedQueueDescriptor
455  = PolymorphicDowncast<const FullyConnectedQueueDescriptor*>(&descriptor);
456  return MakeWorkload<ClFullyConnectedWorkload>(*fullyConnectedQueueDescriptor,
457  info,
458  m_MemoryManager->GetIntraLayerManager(),
459  m_CLCompileContext);
460  }
461  case LayerType::Gather :
462  {
463  auto gatherQueueDescriptor = PolymorphicDowncast<const GatherQueueDescriptor*>(&descriptor);
464  return MakeWorkload<ClGatherWorkload>(*gatherQueueDescriptor, info, m_CLCompileContext);
465  }
466  case LayerType::Input :
467  {
468  auto inputQueueDescriptor = PolymorphicDowncast<const InputQueueDescriptor*>(&descriptor);
469  return std::make_unique<CopyMemGenericWorkload>(*inputQueueDescriptor, info);
470  }
472  {
473  auto instanceNormalizationQueueDescriptor
474  = PolymorphicDowncast<const InstanceNormalizationQueueDescriptor*>(&descriptor);
475  return MakeWorkload<ClInstanceNormalizationWorkload>(*instanceNormalizationQueueDescriptor,
476  info,
477  m_CLCompileContext);
478  }
480  {
481  auto l2NormalizationQueueDescriptor
482  = PolymorphicDowncast<const L2NormalizationQueueDescriptor*>(&descriptor);
483  return MakeWorkload<ClL2NormalizationFloatWorkload, NullWorkload>(*l2NormalizationQueueDescriptor,
484  info,
485  m_CLCompileContext);
486  }
488  {
489  auto logicalBinaryQueueDescriptor = PolymorphicDowncast<const LogicalBinaryQueueDescriptor*>(&descriptor);
490 
491  switch(logicalBinaryQueueDescriptor->m_Parameters.m_Operation)
492  {
494  return std::make_unique<ClLogicalAndWorkload>(*logicalBinaryQueueDescriptor,
495  info,
496  m_CLCompileContext);
498  return std::make_unique<ClLogicalOrWorkload>(*logicalBinaryQueueDescriptor,
499  info,
500  m_CLCompileContext);
501  default:
502  return nullptr;
503  }
504  }
505  case LayerType::LogSoftmax :
506  {
507  auto logSoftmaxQueueDescriptor = PolymorphicDowncast<const LogSoftmaxQueueDescriptor*>(&descriptor);
508 
509  return MakeWorkload<ClLogSoftmaxWorkload>(*logSoftmaxQueueDescriptor,
510  info,
511  m_MemoryManager->GetIntraLayerManager(),
512  m_CLCompileContext);
513  }
514  case LayerType::Lstm :
515  {
516  auto lstmQueueDescriptor = PolymorphicDowncast<const LstmQueueDescriptor*>(&descriptor);
517  return MakeWorkload<ClLstmFloatWorkload, NullWorkload>(*lstmQueueDescriptor, info, m_CLCompileContext);
518  }
519  case LayerType::Maximum :
520  {
521  auto maximumQueueDescriptor = PolymorphicDowncast<const MaximumQueueDescriptor*>(&descriptor);
522  return MakeWorkload<ClMaximumWorkload>(*maximumQueueDescriptor, info, m_CLCompileContext);
523  }
524  case LayerType::Mean :
525  {
526  auto meanQueueDescriptor = PolymorphicDowncast<const MeanQueueDescriptor*>(&descriptor);
527  return MakeWorkload<ClMeanWorkload>(*meanQueueDescriptor, info, m_CLCompileContext);
528  }
529  case LayerType::MemCopy :
530  {
531  auto memCopyQueueDescriptor = PolymorphicDowncast<const MemCopyQueueDescriptor*>(&descriptor);
532  if (memCopyQueueDescriptor->m_Inputs.empty() || !memCopyQueueDescriptor->m_Inputs[0])
533  {
534  throw InvalidArgumentException("ClWorkloadFactory: Invalid null input for MemCopy workload");
535  }
536  return MakeWorkload<CopyMemGenericWorkload>(*memCopyQueueDescriptor, info);
537  }
538  case LayerType::MemImport :
539  {
540  auto memImportQueueDescriptor = PolymorphicDowncast<const MemImportQueueDescriptor*>(&descriptor);
541  if (memImportQueueDescriptor->m_Inputs.empty() || !memImportQueueDescriptor->m_Inputs[0])
542  {
543  throw InvalidArgumentException("ClWorkloadFactory: Invalid null input for MemImport workload");
544  }
545  return std::make_unique<ImportMemGenericWorkload>(*memImportQueueDescriptor, info);
546  }
547  case LayerType::Minimum :
548  {
549  auto minimumQueueDescriptor = PolymorphicDowncast<const MinimumQueueDescriptor*>(&descriptor);
550  return MakeWorkload<ClMinimumWorkload>(*minimumQueueDescriptor, info, m_CLCompileContext);
551  }
553  {
554  auto multiplicationQueueDescriptor = PolymorphicDowncast<const MultiplicationQueueDescriptor*>(&descriptor);
555  return MakeWorkload<ClMultiplicationWorkload>(*multiplicationQueueDescriptor, info, m_CLCompileContext);
556  }
558  {
559  auto normalizationQueueDescriptor = PolymorphicDowncast<const NormalizationQueueDescriptor*>(&descriptor);
560  return MakeWorkload<ClNormalizationFloatWorkload, NullWorkload>(*normalizationQueueDescriptor,
561  info,
562  m_CLCompileContext);
563  }
564  case LayerType::Output :
565  {
566  auto outputQueueDescriptor = PolymorphicDowncast<const OutputQueueDescriptor*>(&descriptor);
567  return std::make_unique<CopyMemGenericWorkload>(*outputQueueDescriptor, info);
568  }
569  case LayerType::Pad :
570  {
571  auto padQueueDescriptor = PolymorphicDowncast<const PadQueueDescriptor*>(&descriptor);
572  return MakeWorkload<ClPadWorkload>(*padQueueDescriptor, info, m_CLCompileContext);
573  }
574  case LayerType::Permute :
575  {
576  auto permuteQueueDescriptor = PolymorphicDowncast<const PermuteQueueDescriptor*>(&descriptor);
577  return MakeWorkload<ClPermuteWorkload>(*permuteQueueDescriptor, info, m_CLCompileContext);
578  }
579  case LayerType::Pooling2d :
580  {
581  auto pooling2dQueueDescriptor = PolymorphicDowncast<const Pooling2dQueueDescriptor*>(&descriptor);
582  return MakeWorkload<ClPooling2dWorkload>(*pooling2dQueueDescriptor, info, m_CLCompileContext);
583  }
585  {
586  auto preCompiledQueueDescriptor = PolymorphicDowncast<const PreCompiledQueueDescriptor*>(&descriptor);
587  return MakeWorkload<NullWorkload, NullWorkload>(*preCompiledQueueDescriptor, info, m_CLCompileContext);
588  }
589  case LayerType::Prelu :
590  {
591  auto preluQueueDescriptor = PolymorphicDowncast<const PreluQueueDescriptor*>(&descriptor);
592  return MakeWorkload<ClPreluWorkload>(*preluQueueDescriptor, info, m_CLCompileContext);
593  }
594  case LayerType::QLstm :
595  {
596  auto qLstmQueueDescriptor = PolymorphicDowncast<const QLstmQueueDescriptor*>(&descriptor);
597  return std::make_unique<ClQLstmWorkload>(*qLstmQueueDescriptor, info, m_CLCompileContext);
598  }
599  case LayerType::Quantize :
600  {
601  auto quantizeQueueDescriptor = PolymorphicDowncast<const QuantizeQueueDescriptor*>(&descriptor);
602  return MakeWorkload<ClQuantizeWorkload>(*quantizeQueueDescriptor, info, m_CLCompileContext);
603  }
605  {
606  auto quantizedLstmQueueDescriptor = PolymorphicDowncast<const QuantizedLstmQueueDescriptor*>(&descriptor);
607  return MakeWorkload<ClQuantizedLstmWorkload>(*quantizedLstmQueueDescriptor, info, m_CLCompileContext);
608  }
609  case LayerType::Rank :
610  {
611  auto rankQueueDescriptor = PolymorphicDowncast<const RankQueueDescriptor*>(&descriptor);
612  return std::make_unique<ClRankWorkload>(*rankQueueDescriptor, info);
613  }
614  case LayerType::Reduce :
615  {
616  auto reduceQueueDescriptor = PolymorphicDowncast<const ReduceQueueDescriptor*>(&descriptor);
617  return std::make_unique<ClReduceWorkload>(*reduceQueueDescriptor, info);
618  }
619  case LayerType::Reshape :
620  {
621  auto reshapeQueueDescriptor = PolymorphicDowncast<const ReshapeQueueDescriptor*>(&descriptor);
622  return MakeWorkload<ClReshapeWorkload>(*reshapeQueueDescriptor, info, m_CLCompileContext);
623  }
624  case LayerType::Resize :
625  {
626  auto resizeQueueDescriptor = PolymorphicDowncast<const ResizeQueueDescriptor*>(&descriptor);
627  return MakeWorkload<ClResizeWorkload>(*resizeQueueDescriptor, info, m_CLCompileContext);
628  }
629  case LayerType::Slice :
630  {
631  auto sliceQueueDescriptor = PolymorphicDowncast<const SliceQueueDescriptor*>(&descriptor);
632  return MakeWorkload<ClSliceWorkload>(*sliceQueueDescriptor, info, m_CLCompileContext);
633  }
634  case LayerType::Softmax :
635  {
636  auto softmaxQueueDescriptor = PolymorphicDowncast<const SoftmaxQueueDescriptor*>(&descriptor);
637  return std::make_unique<ClSoftmaxWorkload>(*softmaxQueueDescriptor,
638  info,
639  m_MemoryManager->GetIntraLayerManager(),
640  m_CLCompileContext);
641  }
643  {
644  auto spaceToBatchNdQueueDescriptor
645  = PolymorphicDowncast<const SpaceToBatchNdQueueDescriptor*>(&descriptor);
646  return MakeWorkload<ClSpaceToBatchNdWorkload>(*spaceToBatchNdQueueDescriptor, info, m_CLCompileContext);
647  }
649  {
650  auto spaceToDepthQueueDescriptor = PolymorphicDowncast<const SpaceToDepthQueueDescriptor*>(&descriptor);
651  return MakeWorkload<ClSpaceToDepthWorkload>(*spaceToDepthQueueDescriptor, info, m_CLCompileContext);
652  }
653  case LayerType::Splitter :
654  {
655  auto splitterQueueDescriptor = PolymorphicDowncast<const SplitterQueueDescriptor*>(&descriptor);
656  return MakeWorkload<ClSplitterWorkload>(*splitterQueueDescriptor, info, m_CLCompileContext);
657  }
658  case LayerType::Stack :
659  {
660  auto stackQueueDescriptor = PolymorphicDowncast<const StackQueueDescriptor*>(&descriptor);
661  return MakeWorkload<ClStackWorkload>(*stackQueueDescriptor, info, m_CLCompileContext);
662  }
664  {
665  auto stridedSliceQueueDescriptor = PolymorphicDowncast<const StridedSliceQueueDescriptor*>(&descriptor);
666  return MakeWorkload<ClStridedSliceWorkload>(*stridedSliceQueueDescriptor, info, m_CLCompileContext);
667  }
669  {
670  auto subtractionQueueDescriptor = PolymorphicDowncast<const SubtractionQueueDescriptor*>(&descriptor);
671  return MakeWorkload<ClSubtractionWorkload>(*subtractionQueueDescriptor, info, m_CLCompileContext);
672  }
673  case LayerType::Transpose :
674  {
675  auto transposeQueueDescriptor = PolymorphicDowncast<const TransposeQueueDescriptor*>(&descriptor);
676  return MakeWorkload<ClTransposeWorkload>(*transposeQueueDescriptor, info, m_CLCompileContext);
677  }
679  {
680  auto transposeConvolution2dQueueDescriptor
681  = PolymorphicDowncast<const TransposeConvolution2dQueueDescriptor*>(&descriptor);
682  return MakeWorkload<ClTransposeConvolution2dWorkload>(*transposeConvolution2dQueueDescriptor,
683  info,
684  m_MemoryManager->GetIntraLayerManager(),
685  m_CLCompileContext);
686  }
687  default:
688  return nullptr;
689  }
690 }
691 
692 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateActivation(const ActivationQueueDescriptor& descriptor,
693  const WorkloadInfo& info) const
694 {
695  return MakeWorkload<ClActivationWorkload>(descriptor, info, m_CLCompileContext);
696 }
697 
698 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateAddition(const AdditionQueueDescriptor& descriptor,
699  const WorkloadInfo& info) const
700 {
701  return MakeWorkload<ClAdditionWorkload>(descriptor, info, m_CLCompileContext);
702 }
703 
704 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateArgMinMax(const ArgMinMaxQueueDescriptor& descriptor,
705  const WorkloadInfo& info) const
706 {
707  return std::make_unique<ClArgMinMaxWorkload>(descriptor, info, m_CLCompileContext);
708 }
709 
710 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateBatchNormalization(
711  const BatchNormalizationQueueDescriptor& descriptor,
712  const WorkloadInfo& info) const
713 {
714  return MakeWorkload<ClBatchNormalizationFloatWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
715 }
716 
717 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor& descriptor,
718  const WorkloadInfo& info) const
719 {
720  return MakeWorkload<ClBatchToSpaceNdWorkload>(descriptor, info, m_CLCompileContext);
721 }
722 
723 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateCast(const CastQueueDescriptor& descriptor,
724  const WorkloadInfo& info) const
725 {
726  return MakeWorkload<ClCastWorkload>(descriptor, info, m_CLCompileContext);
727 }
728 
729 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateChannelShuffle(const ChannelShuffleQueueDescriptor& descriptor,
730  const WorkloadInfo& info) const
731 {
732  return MakeWorkload<ClChannelShuffleWorkload>(descriptor, info, m_CLCompileContext);
733 }
734 
735 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateComparison(const ComparisonQueueDescriptor& descriptor,
736  const WorkloadInfo& info) const
737 {
738  return MakeWorkload<ClComparisonWorkload>(descriptor, info, m_CLCompileContext);
739 }
740 
741 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConcat(const ConcatQueueDescriptor& descriptor,
742  const WorkloadInfo& info) const
743 {
744  return MakeWorkload<ClConcatWorkload>(descriptor, info, m_CLCompileContext);
745 }
746 
747 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConstant(const ConstantQueueDescriptor& descriptor,
748  const WorkloadInfo& info) const
749 {
750  return MakeWorkload<ClConstantWorkload>(descriptor, info, m_CLCompileContext);
751 }
752 
753 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConvertFp16ToFp32(
754  const ConvertFp16ToFp32QueueDescriptor& descriptor,
755  const WorkloadInfo& info) const
756 {
757  return MakeWorkload<ClConvertFp16ToFp32Workload>(descriptor, info, m_CLCompileContext);
758 }
759 
760 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConvertFp32ToFp16(
761  const ConvertFp32ToFp16QueueDescriptor& descriptor,
762  const WorkloadInfo& info) const
763 {
764  return MakeWorkload<ClConvertFp32ToFp16Workload>(descriptor, info, m_CLCompileContext);
765 }
766 
767 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConvolution2d(const Convolution2dQueueDescriptor& descriptor,
768  const WorkloadInfo& info) const
769 {
770  bool isFastMathEnabled = false;
771  if (m_ModelContextPtr)
772  {
773  if (m_ModelContextPtr.get() != nullptr)
774  {
775  auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
776  if (modelOptions)
777  {
778  isFastMathEnabled = modelOptions->IsFastMathEnabled();
779  }
780  }
781  }
782  return MakeWorkload<ClConvolution2dWorkload>(descriptor,
783  info,
784  m_MemoryManager->GetIntraLayerManager(),
785  m_CLCompileContext,
786  isFastMathEnabled);
787 }
788 
789 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateConvolution3d(const Convolution3dQueueDescriptor& descriptor,
790  const WorkloadInfo& info) const
791 {
792  bool isFastMathEnabled = false;
793  if (m_ModelContextPtr)
794  {
795  if (m_ModelContextPtr.get() != nullptr)
796  {
797  auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
798  if (modelOptions)
799  {
800  isFastMathEnabled = modelOptions->IsFastMathEnabled();
801  }
802  }
803  }
804  return MakeWorkload<ClConvolution3dWorkload>(descriptor,
805  info,
806  m_MemoryManager->GetIntraLayerManager(),
807  m_CLCompileContext,
808  isFastMathEnabled);
809 }
810 
811 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDebug(const DebugQueueDescriptor& descriptor,
812  const WorkloadInfo& info) const
813 {
814  return MakeWorkload<NullWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
815 }
816 
817 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDepthToSpace(const DepthToSpaceQueueDescriptor& descriptor,
818  const WorkloadInfo& info) const
819 {
820  return MakeWorkload<ClDepthToSpaceWorkload>(descriptor, info, m_CLCompileContext);
821 }
822 
823 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDepthwiseConvolution2d(
824  const DepthwiseConvolution2dQueueDescriptor& descriptor,
825  const WorkloadInfo& info) const
826 {
827  return MakeWorkload<ClDepthwiseConvolutionWorkload>(descriptor, info, m_CLCompileContext);
828 }
829 
830 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDequantize(const DequantizeQueueDescriptor& descriptor,
831  const WorkloadInfo& info) const
832 {
833  return MakeWorkload<ClDequantizeWorkload>(descriptor, info, m_CLCompileContext);
834 }
835 
836 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDetectionPostProcess(
837  const DetectionPostProcessQueueDescriptor& descriptor,
838  const WorkloadInfo& info) const
839 {
840  return MakeWorkload<NullWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
841 }
842 
843 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateDivision(const DivisionQueueDescriptor& descriptor,
844  const WorkloadInfo& info) const
845 {
846  return std::make_unique<ClDivisionWorkload>(descriptor, info, m_CLCompileContext);
847 }
848 
849 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateElementwiseUnary(const ElementwiseUnaryQueueDescriptor& descriptor,
850  const WorkloadInfo& info) const
851 {
852  switch(descriptor.m_Parameters.m_Operation)
853  {
854  case UnaryOperation::Abs:
855  {
856  AbsQueueDescriptor absQueueDescriptor;
857  absQueueDescriptor.m_Inputs = descriptor.m_Inputs;
858  absQueueDescriptor.m_Outputs = descriptor.m_Outputs;
859 
860  return std::make_unique<ClAbsWorkload>(absQueueDescriptor, info, m_CLCompileContext);
861  }
862  case UnaryOperation::Exp:
863  return std::make_unique<ClExpWorkload>(descriptor, info, m_CLCompileContext);
864  case UnaryOperation::Log:
865  return std::make_unique<ClLogWorkload>(descriptor, info, m_CLCompileContext);
867  return std::make_unique<ClLogicalNotWorkload>(descriptor, info, m_CLCompileContext);
868  case UnaryOperation::Neg:
869  return std::make_unique<ClNegWorkload>(descriptor, info, m_CLCompileContext);
871  {
872  RsqrtQueueDescriptor rsqrtQueueDescriptor;
873  rsqrtQueueDescriptor.m_Inputs = descriptor.m_Inputs;
874  rsqrtQueueDescriptor.m_Outputs = descriptor.m_Outputs;
875 
876  return std::make_unique<ClRsqrtWorkload>(rsqrtQueueDescriptor, info, m_CLCompileContext);
877  }
878  case UnaryOperation::Sin:
879  return std::make_unique<ClSinWorkload>(descriptor, info, m_CLCompileContext);
880  default:
881  return nullptr;
882  }
883 }
884 
885 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateFill(const FillQueueDescriptor& descriptor,
886  const WorkloadInfo& info) const
887 {
888  return std::make_unique<ClFillWorkload>(descriptor, info, m_CLCompileContext);
889 }
890 
891 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateFloor(const FloorQueueDescriptor& descriptor,
892  const WorkloadInfo& info) const
893 {
894  return MakeWorkload<ClFloorFloatWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
895 }
896 
897 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateFullyConnected(const FullyConnectedQueueDescriptor& descriptor,
898  const WorkloadInfo& info) const
899 {
900  return MakeWorkload<ClFullyConnectedWorkload>(descriptor,
901  info,
902  m_MemoryManager->GetIntraLayerManager(),
903  m_CLCompileContext);
904 }
905 
906 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateGather(const GatherQueueDescriptor& descriptor,
907  const WorkloadInfo& info) const
908 {
909  return MakeWorkload<ClGatherWorkload>(descriptor, info, m_CLCompileContext);
910 }
911 
912 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateInput(const InputQueueDescriptor& descriptor,
913  const WorkloadInfo& info) const
914 {
915  return std::make_unique<CopyMemGenericWorkload>(descriptor, info);
916 }
917 
918 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateInstanceNormalization(
919  const InstanceNormalizationQueueDescriptor& descriptor,
920  const WorkloadInfo& info) const
921 {
922  return MakeWorkload<ClInstanceNormalizationWorkload>(descriptor, info, m_CLCompileContext);
923 }
924 
925 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateL2Normalization(const L2NormalizationQueueDescriptor& descriptor,
926  const WorkloadInfo& info) const
927 {
928  return MakeWorkload<ClL2NormalizationFloatWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
929 }
930 
931 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateLogicalBinary(const LogicalBinaryQueueDescriptor& descriptor,
932  const WorkloadInfo& info) const
933 {
934  switch(descriptor.m_Parameters.m_Operation)
935  {
937  return std::make_unique<ClLogicalAndWorkload>(descriptor, info, m_CLCompileContext);
939  return std::make_unique<ClLogicalOrWorkload>(descriptor, info, m_CLCompileContext);
940  default:
941  return nullptr;
942  }
943 }
944 
945 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateLogSoftmax(const LogSoftmaxQueueDescriptor& descriptor,
946  const WorkloadInfo& info) const
947 {
948  return MakeWorkload<ClLogSoftmaxWorkload>(descriptor,
949  info,
950  m_MemoryManager->GetIntraLayerManager(),
951  m_CLCompileContext);
952 }
953 
954 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateLstm(const LstmQueueDescriptor& descriptor,
955  const WorkloadInfo& info) const
956 {
957  return MakeWorkload<ClLstmFloatWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
958 }
959 
960 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMaximum(const MaximumQueueDescriptor& descriptor,
961  const WorkloadInfo& info) const
962 {
963  return MakeWorkload<ClMaximumWorkload>(descriptor, info, m_CLCompileContext);
964 }
965 
966 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMean(const MeanQueueDescriptor& descriptor,
967  const WorkloadInfo& info) const
968 {
969  return MakeWorkload<ClMeanWorkload>(descriptor, info, m_CLCompileContext);
970 }
971 
972 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMemCopy(const MemCopyQueueDescriptor& descriptor,
973  const WorkloadInfo& info) const
974 {
975  if (descriptor.m_Inputs.empty() || !descriptor.m_Inputs[0])
976  {
977  throw InvalidArgumentException("ClWorkloadFactory: Invalid null input for MemCopy workload");
978  }
979 
980  return MakeWorkload<CopyMemGenericWorkload>(descriptor, info);
981 }
982 
983 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMemImport(const MemImportQueueDescriptor& descriptor,
984  const WorkloadInfo& info) const
985 {
986  if (descriptor.m_Inputs.empty() || !descriptor.m_Inputs[0])
987  {
988  throw InvalidArgumentException("ClWorkloadFactory: Invalid null input for MemImport workload");
989  }
990 
991  return std::make_unique<ImportMemGenericWorkload>(descriptor, info);
992 }
993 
994 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMinimum(const MinimumQueueDescriptor& descriptor,
995  const WorkloadInfo& info) const
996 {
997  return MakeWorkload<ClMinimumWorkload>(descriptor, info, m_CLCompileContext);
998 }
999 
1000 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateMultiplication(const MultiplicationQueueDescriptor& descriptor,
1001  const WorkloadInfo& info) const
1002 {
1003  return MakeWorkload<ClMultiplicationWorkload>(descriptor, info, m_CLCompileContext);
1004 }
1005 
1006 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateNormalization(const NormalizationQueueDescriptor& descriptor,
1007  const WorkloadInfo& info) const
1008 {
1009  return MakeWorkload<ClNormalizationFloatWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
1010 }
1011 
1012 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateOutput(const OutputQueueDescriptor& descriptor,
1013  const WorkloadInfo& info) const
1014 {
1015  return std::make_unique<CopyMemGenericWorkload>(descriptor, info);
1016 }
1017 
1018 std::unique_ptr<IWorkload> ClWorkloadFactory::CreatePad(const PadQueueDescriptor& descriptor,
1019  const WorkloadInfo& info) const
1020 {
1021  return MakeWorkload<ClPadWorkload>(descriptor, info, m_CLCompileContext);
1022 }
1023 
1024 std::unique_ptr<IWorkload> ClWorkloadFactory::CreatePermute(const PermuteQueueDescriptor& descriptor,
1025  const WorkloadInfo& info) const
1026 {
1027  return MakeWorkload<ClPermuteWorkload>(descriptor, info, m_CLCompileContext);
1028 }
1029 
1030 std::unique_ptr<IWorkload> ClWorkloadFactory::CreatePooling2d(const Pooling2dQueueDescriptor& descriptor,
1031  const WorkloadInfo& info) const
1032 {
1033  return MakeWorkload<ClPooling2dWorkload>(descriptor, info, m_CLCompileContext);
1034 }
1035 
1036 std::unique_ptr<IWorkload> ClWorkloadFactory::CreatePreCompiled(const PreCompiledQueueDescriptor& descriptor,
1037  const WorkloadInfo& info) const
1038 {
1039  return MakeWorkload<NullWorkload, NullWorkload>(descriptor, info, m_CLCompileContext);
1040 }
1041 
1042 std::unique_ptr<IWorkload> ClWorkloadFactory::CreatePrelu(const PreluQueueDescriptor &descriptor,
1043  const WorkloadInfo &info) const
1044 {
1045  return MakeWorkload<ClPreluWorkload>(descriptor, info, m_CLCompileContext);
1046 }
1047 
1048 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateQLstm(const QLstmQueueDescriptor& descriptor,
1049  const WorkloadInfo& info) const
1050 {
1051  return std::make_unique<ClQLstmWorkload>(descriptor, info, m_CLCompileContext);
1052 }
1053 
1054 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateQuantize(const QuantizeQueueDescriptor& descriptor,
1055  const WorkloadInfo& info) const
1056 {
1057  return MakeWorkload<ClQuantizeWorkload>(descriptor, info, m_CLCompileContext);
1058 }
1059 
1060 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateQuantizedLstm(const QuantizedLstmQueueDescriptor& descriptor,
1061  const WorkloadInfo& info) const
1062 {
1063  return MakeWorkload<ClQuantizedLstmWorkload>(descriptor, info, m_CLCompileContext);
1064 }
1065 
1066 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateRank(const RankQueueDescriptor& descriptor,
1067  const WorkloadInfo& info) const
1068 {
1069  return std::make_unique<ClRankWorkload>(descriptor, info);
1070 }
1071 
1072 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateReduce(const ReduceQueueDescriptor& descriptor,
1073  const WorkloadInfo& info) const
1074 {
1075  return std::make_unique<ClReduceWorkload>(descriptor, info);
1076 }
1077 
1078 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateReshape(const ReshapeQueueDescriptor& descriptor,
1079  const WorkloadInfo& info) const
1080 {
1081  return MakeWorkload<ClReshapeWorkload>(descriptor, info, m_CLCompileContext);
1082 }
1083 
1084 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateResize(const ResizeQueueDescriptor& descriptor,
1085  const WorkloadInfo& info) const
1086 {
1087  return MakeWorkload<ClResizeWorkload>(descriptor, info, m_CLCompileContext);
1088 }
1089 
1090 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSlice(const SliceQueueDescriptor& descriptor,
1091  const WorkloadInfo& info) const
1092 {
1093  return MakeWorkload<ClSliceWorkload>(descriptor, info, m_CLCompileContext);
1094 }
1095 
1096 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& descriptor,
1097  const WorkloadInfo& info) const
1098 {
1099  return std::make_unique<ClSoftmaxWorkload>(descriptor,
1100  info,
1101  m_MemoryManager->GetIntraLayerManager(),
1102  m_CLCompileContext);
1103 }
1104 
1105 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSpaceToBatchNd(const SpaceToBatchNdQueueDescriptor& descriptor,
1106  const WorkloadInfo& info) const
1107 {
1108  return MakeWorkload<ClSpaceToBatchNdWorkload>(descriptor, info, m_CLCompileContext);
1109 }
1110 
1111 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSpaceToDepth(const SpaceToDepthQueueDescriptor& descriptor,
1112  const WorkloadInfo& info) const
1113 {
1114  return MakeWorkload<ClSpaceToDepthWorkload>(descriptor, info, m_CLCompileContext);
1115 }
1116 
1117 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSplitter(const SplitterQueueDescriptor& descriptor,
1118  const WorkloadInfo& info) const
1119 {
1120  return MakeWorkload<ClSplitterWorkload>(descriptor, info, m_CLCompileContext);
1121 }
1122 
1123 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateStack(const StackQueueDescriptor& descriptor,
1124  const WorkloadInfo& info) const
1125 {
1126  return MakeWorkload<ClStackWorkload>(descriptor, info, m_CLCompileContext);
1127 }
1128 
1129 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateStridedSlice(const StridedSliceQueueDescriptor& descriptor,
1130  const WorkloadInfo& info) const
1131 {
1132  return MakeWorkload<ClStridedSliceWorkload>(descriptor, info, m_CLCompileContext);
1133 }
1134 
1135 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSubtraction(const SubtractionQueueDescriptor& descriptor,
1136  const WorkloadInfo& info) const
1137 {
1138  return MakeWorkload<ClSubtractionWorkload>(descriptor, info, m_CLCompileContext);
1139 }
1140 
1141 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateTranspose(const TransposeQueueDescriptor& descriptor,
1142  const WorkloadInfo& info) const
1143 {
1144  return MakeWorkload<ClTransposeWorkload>(descriptor, info, m_CLCompileContext);
1145 }
1146 
1147 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateTransposeConvolution2d(
1148  const TransposeConvolution2dQueueDescriptor& descriptor,
1149  const WorkloadInfo& info) const
1150 {
1151  return MakeWorkload<ClTransposeConvolution2dWorkload>(descriptor,
1152  info,
1153  m_MemoryManager->GetIntraLayerManager(),
1154  m_CLCompileContext);
1155 }
1156 
1157 } // namespace armnn
std::unique_ptr< IWorkload > CreateSubtraction(const SubtractionQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< ITensorHandle > CreateSubTensorHandle(ITensorHandle &parent, TensorShape const &subTensorShape, unsigned int const *subTensorOrigin) const override
std::unique_ptr< IWorkload > CreateElementwiseUnary(const ElementwiseUnaryQueueDescriptor &descriptor, const WorkloadInfo &info) const override
ClWorkloadFactory(const std::shared_ptr< ClMemoryManager > &memoryManager)
UnaryOperation m_Operation
Specifies the elementwiseUnary operation to execute.
std::unique_ptr< IWorkload > CreateStridedSlice(const StridedSliceQueueDescriptor &, const WorkloadInfo &) const override
Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.
Definition: INetwork.hpp:66
DataLayout
Definition: Types.hpp:49
std::unique_ptr< IWorkload > CreateInput(const InputQueueDescriptor &, const WorkloadInfo &) const override
void AfterWorkloadsCreated() override
std::unique_ptr< IWorkload > CreateL2Normalization(const L2NormalizationQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateBatchNormalization(const BatchNormalizationQueueDescriptor &, const WorkloadInfo &) const override
constexpr const char * ClBackendId()
Definition: ClBackendId.hpp:10
std::vector< BackendOptions > ModelOptions
std::unique_ptr< IWorkload > CreateSoftmax(const SoftmaxQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateSpaceToBatchNd(const SpaceToBatchNdQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreatePooling2d(const Pooling2dQueueDescriptor &, const WorkloadInfo &) const override
std::array< unsigned int, MaxNumOfTensorDimensions > Coordinates
std::unique_ptr< IWorkload > CreateMean(const MeanQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreatePad(const PadQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &tensorInfo, const bool IsMemoryManaged=true) const override
std::unique_ptr< IWorkload > CreateLstm(const LstmQueueDescriptor &, const WorkloadInfo &) const override
const BackendId & GetBackendId() const override
std::unique_ptr< IWorkload > CreateDepthToSpace(const DepthToSpaceQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateLogSoftmax(const LogSoftmaxQueueDescriptor &, const WorkloadInfo &) const override
#define ARMNN_LOG(severity)
Definition: Logging.hpp:205
std::unique_ptr< IWorkload > CreateComparison(const ComparisonQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateTranspose(const TransposeQueueDescriptor &, const WorkloadInfo &) const override
bool SaveSerializedToStream(std::ostream &stream)
Serializes the ClContext to the stream.
Copyright (c) 2021 ARM Limited and Contributors.
void IgnoreUnused(Ts &&...)
TypedWorkload< QueueDescriptor, armnn::DataType::Float16, armnn::DataType::Float32 > FloatWorkload
Definition: Workload.hpp:189
std::unique_ptr< IWorkload > CreateDequantize(const DequantizeQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateSplitter(const SplitterQueueDescriptor &, const WorkloadInfo &) const override
LogicalBinaryOperation m_Operation
Specifies the logical operation to execute.
std::unique_ptr< IWorkload > CreateConvertFp32ToFp16(const ConvertFp32ToFp16QueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateNormalization(const NormalizationQueueDescriptor &, const WorkloadInfo &) const override
void DeserializeFromBinary(arm_compute::CLCompileContext &clCompileContext, cl::Context &context, cl::Device &device, const std::vector< uint8_t > &binaryContent)
Deserializes the CLCompileContext built-in programs from binary file contents.
std::unique_ptr< IWorkload > CreateDepthwiseConvolution2d(const DepthwiseConvolution2dQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateInstanceNormalization(const InstanceNormalizationQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateTransposeConvolution2d(const TransposeConvolution2dQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateSpaceToDepth(const SpaceToDepthQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateConcat(const ConcatQueueDescriptor &, const WorkloadInfo &) const override
static bool IsLayerSupported(const Layer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
std::unique_ptr< IWorkload > CreateWorkload(LayerType type, const QueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload > CreatePrelu(const PreluQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateArgMinMax(const ArgMinMaxQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreatePermute(const PermuteQueueDescriptor &, const WorkloadInfo &) const override
std::shared_ptr< IBackendModelContext > IBackendSpecificModelContextPtr
std::unique_ptr< IWorkload > CreateDebug(const DebugQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateConstant(const ConstantQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateConvolution2d(const Convolution2dQueueDescriptor &, const WorkloadInfo &) const override
void Serialize(const arm_compute::CLCompileContext &clCompileContext)
Serializes the CLCompileContext built-in programs.
std::unique_ptr< IWorkload > CreateFullyConnected(const FullyConnectedQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateMinimum(const MinimumQueueDescriptor &, const WorkloadInfo &) const override
RuntimeException WrapClError(const cl::Error &clError, const CheckLocation &location)
std::unique_ptr< IWorkload > CreateReshape(const ReshapeQueueDescriptor &, const WorkloadInfo &) const override
static bool IsLayerSupported(const BackendId &backendId, const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
std::unique_ptr< IWorkload > CreateFloor(const FloorQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateRank(const RankQueueDescriptor &, const WorkloadInfo &) const override
#define CHECK_LOCATION()
Definition: Exceptions.hpp:209
virtual TensorShape GetShape() const =0
Get the number of elements for each dimension ordered from slowest iterating dimension to fastest ite...
std::unique_ptr< IWorkload > CreateDetectionPostProcess(const DetectionPostProcessQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateQuantize(const QuantizeQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateAddition(const AdditionQueueDescriptor &, const WorkloadInfo &) const override
void Deserialize(arm_compute::CLCompileContext &clCompileContext, cl::Context &context, cl::Device &device, const std::string &filePath)
Deserializes the CLCompileContext built-in programs from a binary file.
TypedWorkload< QueueDescriptor, armnn::DataType::QAsymmU8 > Uint8Workload
Definition: Workload.hpp:195
std::unique_ptr< IWorkload > CreateGather(const GatherQueueDescriptor &, const WorkloadInfo &) const override
std::vector< ITensorHandle * > m_Outputs
std::unique_ptr< IWorkload > CreateDivision(const DivisionQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreatePreCompiled(const PreCompiledQueueDescriptor &, const WorkloadInfo &) const override
unsigned int GetNumDimensions() const
Function that returns the tensor rank.
Definition: Tensor.cpp:174
std::unique_ptr< IWorkload > CreateStack(const StackQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateOutput(const OutputQueueDescriptor &, const WorkloadInfo &) const override
Contains information about TensorInfos of a layer.
std::unique_ptr< IWorkload > CreateResize(const ResizeQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateSlice(const SliceQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateConvertFp16ToFp32(const ConvertFp16ToFp32QueueDescriptor &, const WorkloadInfo &) const override
std::vector< ITensorHandle * > m_Inputs
std::unique_ptr< IWorkload > CreateMultiplication(const MultiplicationQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateMaximum(const MaximumQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateActivation(const ActivationQueueDescriptor &, const WorkloadInfo &) const override
The ClBackendModelContext is used to pass in CL specific backend ModelOptions.
std::unique_ptr< IWorkload > CreateMemImport(const MemImportQueueDescriptor &, const WorkloadInfo &) const override
std::string GetCachedNetworkFilePath() const
std::unique_ptr< IWorkload > CreateMemCopy(const MemCopyQueueDescriptor &, const WorkloadInfo &) const override
Depthwise Convolution 2D layer workload data.
std::unique_ptr< IWorkload > CreateQuantizedLstm(const QuantizedLstmQueueDescriptor &, const WorkloadInfo &) const override
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below...
Definition: Types.hpp:458