xref: /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/ClElementwiseUnary.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/ClElementwiseUnary.h"
25 
26 #include "src/gpu/cl/kernels/ClElementwiseUnaryKernel.h"
27 
28 #include "src/common/utils/Log.h"
29 
30 namespace arm_compute
31 {
32 namespace opencl
33 {
configure(const ClCompileContext & compile_context,const ITensorInfo * src,ITensorInfo * dst)34 void ClRsqrt::configure(const ClCompileContext &compile_context, const ITensorInfo *src, ITensorInfo *dst)
35 {
36     ARM_COMPUTE_LOG_PARAMS(src, dst);
37     auto k = std::make_unique<kernels::ClElementWiseUnaryKernel>();
38     k->configure(compile_context, src, dst, ElementWiseUnary::RSQRT);
39     _kernel = std::move(k);
40 }
41 
validate(const ITensorInfo * src,const ITensorInfo * dst)42 Status ClRsqrt::validate(const ITensorInfo *src, const ITensorInfo *dst)
43 {
44     return kernels::ClElementWiseUnaryKernel::validate(src, dst, ElementWiseUnary::RSQRT);
45 }
46 
configure(const ClCompileContext & compile_context,const ITensorInfo * src,ITensorInfo * dst)47 void ClExp::configure(const ClCompileContext &compile_context, const ITensorInfo *src, ITensorInfo *dst)
48 {
49     ARM_COMPUTE_LOG_PARAMS(src, dst);
50     auto k = std::make_unique<kernels::ClElementWiseUnaryKernel>();
51     k->configure(compile_context, src, dst, ElementWiseUnary::EXP);
52     _kernel = std::move(k);
53 }
54 
validate(const ITensorInfo * src,const ITensorInfo * dst)55 Status ClExp::validate(const ITensorInfo *src, const ITensorInfo *dst)
56 {
57     return kernels::ClElementWiseUnaryKernel::validate(src, dst, ElementWiseUnary::EXP);
58 }
59 
configure(const ClCompileContext & compile_context,const ITensorInfo * src,ITensorInfo * dst)60 void ClNeg::configure(const ClCompileContext &compile_context, const ITensorInfo *src, ITensorInfo *dst)
61 {
62     ARM_COMPUTE_LOG_PARAMS(src, dst);
63     auto k = std::make_unique<kernels::ClElementWiseUnaryKernel>();
64     k->configure(compile_context, src, dst, ElementWiseUnary::NEG);
65     _kernel = std::move(k);
66 }
67 
validate(const ITensorInfo * src,const ITensorInfo * dst)68 Status ClNeg::validate(const ITensorInfo *src, const ITensorInfo *dst)
69 {
70     return kernels::ClElementWiseUnaryKernel::validate(src, dst, ElementWiseUnary::NEG);
71 }
72 
configure(const ClCompileContext & compile_context,const ITensorInfo * src,ITensorInfo * dst)73 void ClSin::configure(const ClCompileContext &compile_context, const ITensorInfo *src, ITensorInfo *dst)
74 {
75     ARM_COMPUTE_LOG_PARAMS(src, dst);
76     auto k = std::make_unique<kernels::ClElementWiseUnaryKernel>();
77     k->configure(compile_context, src, dst, ElementWiseUnary::SIN);
78     _kernel = std::move(k);
79 }
80 
validate(const ITensorInfo * src,const ITensorInfo * dst)81 Status ClSin::validate(const ITensorInfo *src, const ITensorInfo *dst)
82 {
83     return kernels::ClElementWiseUnaryKernel::validate(src, dst, ElementWiseUnary::SIN);
84 }
85 
configure(const ClCompileContext & compile_context,const ITensorInfo * src,ITensorInfo * dst)86 void ClAbs::configure(const ClCompileContext &compile_context, const ITensorInfo *src, ITensorInfo *dst)
87 {
88     ARM_COMPUTE_LOG_PARAMS(src, dst);
89     auto k = std::make_unique<kernels::ClElementWiseUnaryKernel>();
90     k->configure(compile_context, src, dst, ElementWiseUnary::ABS);
91     _kernel = std::move(k);
92 }
93 
validate(const ITensorInfo * src,const ITensorInfo * dst)94 Status ClAbs::validate(const ITensorInfo *src, const ITensorInfo *dst)
95 {
96     return kernels::ClElementWiseUnaryKernel::validate(src, dst, ElementWiseUnary::ABS);
97 }
98 
configure(const ClCompileContext & compile_context,const ITensorInfo * src,ITensorInfo * dst)99 void ClLog::configure(const ClCompileContext &compile_context, const ITensorInfo *src, ITensorInfo *dst)
100 {
101     ARM_COMPUTE_LOG_PARAMS(src, dst);
102     auto k = std::make_unique<kernels::ClElementWiseUnaryKernel>();
103     k->configure(compile_context, src, dst, ElementWiseUnary::LOG);
104     _kernel = std::move(k);
105 }
106 
validate(const ITensorInfo * src,const ITensorInfo * dst)107 Status ClLog::validate(const ITensorInfo *src, const ITensorInfo *dst)
108 {
109     return kernels::ClElementWiseUnaryKernel::validate(src, dst, ElementWiseUnary::LOG);
110 }
111 
configure(const ClCompileContext & compile_context,const ITensorInfo * src,ITensorInfo * dst)112 void ClRound::configure(const ClCompileContext &compile_context, const ITensorInfo *src, ITensorInfo *dst)
113 {
114     ARM_COMPUTE_LOG_PARAMS(src, dst);
115     auto k = std::make_unique<kernels::ClElementWiseUnaryKernel>();
116     k->configure(compile_context, src, dst, ElementWiseUnary::ROUND);
117     _kernel = std::move(k);
118 }
119 
validate(const ITensorInfo * src,const ITensorInfo * dst)120 Status ClRound::validate(const ITensorInfo *src, const ITensorInfo *dst)
121 {
122     return kernels::ClElementWiseUnaryKernel::validate(src, dst, ElementWiseUnary::ROUND);
123 }
124 } // namespace opencl
125 } // namespace arm_compute
126