xref: /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/ClElementwiseOperations.cpp (revision c217d954acce2dbc11938adb493fc0abd69584f3)
1 /*
2  * Copyright (c) 2021 Arm Limited.
3  *
4  * SPDX-License-Identifier: MIT
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to
8  * deal in the Software without restriction, including without limitation the
9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10  * sell copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
24 #include "src/gpu/cl/operators/ClElementwiseOperations.h"
25 
26 #include "src/gpu/cl/kernels/ClElementwiseKernel.h"
27 
28 #include "src/common/utils/Log.h"
29 
30 namespace arm_compute
31 {
32 namespace opencl
33 {
configure(const ClCompileContext & compile_context,ITensorInfo * src1,ITensorInfo * src2,ITensorInfo * dst,const ActivationLayerInfo & act_info)34 void ClElementwiseDivision::configure(const ClCompileContext &compile_context, ITensorInfo *src1, ITensorInfo *src2, ITensorInfo *dst, const ActivationLayerInfo &act_info)
35 {
36     ARM_COMPUTE_LOG_PARAMS(src1, src2, dst, act_info);
37     auto k = std::make_unique<kernels::ClArithmeticKernel>();
38     k->configure(compile_context, ArithmeticOperation::DIV, src1, src2, dst, act_info);
39     _kernel = std::move(k);
40 }
41 
validate(const ITensorInfo * src1,const ITensorInfo * src2,const ITensorInfo * dst,const ActivationLayerInfo & act_info)42 Status ClElementwiseDivision::validate(const ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const ActivationLayerInfo &act_info)
43 {
44     return kernels::ClArithmeticKernel::validate(ArithmeticOperation::DIV, src1, src2, dst, act_info);
45 }
46 
configure(const ClCompileContext & compile_context,ITensorInfo * src1,ITensorInfo * src2,ITensorInfo * dst,const ActivationLayerInfo & act_info)47 void ClElementwiseMax::configure(const ClCompileContext &compile_context, ITensorInfo *src1, ITensorInfo *src2, ITensorInfo *dst, const ActivationLayerInfo &act_info)
48 {
49     ARM_COMPUTE_LOG_PARAMS(src1, src2, dst, act_info);
50     auto k = std::make_unique<kernels::ClArithmeticKernel>();
51     k->configure(compile_context, ArithmeticOperation::MAX, src1, src2, dst, act_info);
52     _kernel = std::move(k);
53 }
54 
validate(const ITensorInfo * src1,const ITensorInfo * src2,const ITensorInfo * dst,const ActivationLayerInfo & act_info)55 Status ClElementwiseMax::validate(const ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const ActivationLayerInfo &act_info)
56 {
57     return kernels::ClArithmeticKernel::validate(ArithmeticOperation::MAX, src1, src2, dst, act_info);
58 }
59 
configure(const ClCompileContext & compile_context,ITensorInfo * src1,ITensorInfo * src2,ITensorInfo * dst,const ActivationLayerInfo & act_info)60 void ClElementwiseMin::configure(const ClCompileContext &compile_context, ITensorInfo *src1, ITensorInfo *src2, ITensorInfo *dst, const ActivationLayerInfo &act_info)
61 {
62     ARM_COMPUTE_LOG_PARAMS(src1, src2, dst, act_info);
63     auto k = std::make_unique<kernels::ClArithmeticKernel>();
64     k->configure(compile_context, ArithmeticOperation::MIN, src1, src2, dst, act_info);
65     _kernel = std::move(k);
66 }
67 
validate(const ITensorInfo * src1,const ITensorInfo * src2,const ITensorInfo * dst,const ActivationLayerInfo & act_info)68 Status ClElementwiseMin::validate(const ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const ActivationLayerInfo &act_info)
69 {
70     return kernels::ClArithmeticKernel::validate(ArithmeticOperation::MIN, src1, src2, dst, act_info);
71 }
72 
configure(const ClCompileContext & compile_context,ITensorInfo * src1,ITensorInfo * src2,ITensorInfo * dst,const ActivationLayerInfo & act_info)73 void ClElementwiseSquaredDiff::configure(const ClCompileContext &compile_context, ITensorInfo *src1, ITensorInfo *src2, ITensorInfo *dst, const ActivationLayerInfo &act_info)
74 {
75     ARM_COMPUTE_LOG_PARAMS(src1, src2, dst, act_info);
76     auto k = std::make_unique<kernels::ClArithmeticKernel>();
77     k->configure(compile_context, ArithmeticOperation::SQUARED_DIFF, src1, src2, dst, act_info);
78     _kernel = std::move(k);
79 }
80 
validate(const ITensorInfo * src1,const ITensorInfo * src2,const ITensorInfo * dst,const ActivationLayerInfo & act_info)81 Status ClElementwiseSquaredDiff::validate(const ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const ActivationLayerInfo &act_info)
82 {
83     return kernels::ClArithmeticKernel::validate(ArithmeticOperation::SQUARED_DIFF, src1, src2, dst, act_info);
84 }
85 
configure(const ClCompileContext & compile_context,ITensorInfo * src1,ITensorInfo * src2,ITensorInfo * dst,const ActivationLayerInfo & act_info)86 void ClElementwisePower::configure(const ClCompileContext &compile_context, ITensorInfo *src1, ITensorInfo *src2, ITensorInfo *dst, const ActivationLayerInfo &act_info)
87 {
88     ARM_COMPUTE_LOG_PARAMS(src1, src2, dst, act_info);
89     auto k = std::make_unique<kernels::ClArithmeticKernel>();
90     k->configure(compile_context, ArithmeticOperation::POWER, src1, src2, dst, act_info);
91     _kernel = std::move(k);
92 }
93 
validate(const ITensorInfo * src1,const ITensorInfo * src2,const ITensorInfo * dst,const ActivationLayerInfo & act_info)94 Status ClElementwisePower::validate(const ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const ActivationLayerInfo &act_info)
95 {
96     return kernels::ClArithmeticKernel::validate(ArithmeticOperation::POWER, src1, src2, dst, act_info);
97 }
98 } // namespace opencl
99 } // namespace arm_compute
100