diff options
Diffstat (limited to 'src/backends/cl/ClWorkloadFactory.cpp')
-rw-r--r-- | src/backends/cl/ClWorkloadFactory.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/cl/ClWorkloadFactory.cpp b/src/backends/cl/ClWorkloadFactory.cpp index 022867710c..493080f7af 100644 --- a/src/backends/cl/ClWorkloadFactory.cpp +++ b/src/backends/cl/ClWorkloadFactory.cpp @@ -459,6 +459,13 @@ std::unique_ptr<IWorkload> ClWorkloadFactory::CreateWorkload(LayerType type, info, m_CLCompileContext); } + case BinaryOperation::Power: + case BinaryOperation::SqDiff: + { + return std::make_unique<ClElementwiseBinaryWorkload>(*elementwiseBinaryQueueDescriptor, + info, + m_CLCompileContext); + } case BinaryOperation::Sub: { SubtractionQueueDescriptor subtractionQueueDescriptor; |