blob: 137a0de6a70ab905969e94b569f7ab78147df1f7 [file] [log] [blame]
Sheri Zhangf9ab9f92021-03-16 12:09:15 +00001/*
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/runtime/gpu/cl/operators/ClPixelWiseMultiplication.h"
25
26#include "arm_compute/runtime/CL/CLScheduler.h"
Sheri Zhangf9ab9f92021-03-16 12:09:15 +000027#include "src/core/gpu/cl/ClCompileContext.h"
28#include "src/core/gpu/cl/kernels/ClPixelWiseMultiplicationKernel.h"
29
30namespace arm_compute
31{
32namespace opencl
33{
Sheri Zhangf9ab9f92021-03-16 12:09:15 +000034void ClPixelWiseMultiplication::configure(const CLCompileContext &compile_context, ITensorInfo *src1, ITensorInfo *src2, ITensorInfo *dst, float scale,
35 ConvertPolicy overflow_policy, RoundingPolicy rounding_policy, const ActivationLayerInfo &act_info)
36{
37 auto k = std::make_unique<kernels::ClPixelWiseMultiplicationKernel>();
38 k->configure(compile_context, src1, src2, dst, scale, overflow_policy, rounding_policy, act_info);
39 _kernel = std::move(k);
Sheri Zhangf9ab9f92021-03-16 12:09:15 +000040}
41
42Status ClPixelWiseMultiplication::validate(const ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, float scale,
43 ConvertPolicy overflow_policy, RoundingPolicy rounding_policy, const ActivationLayerInfo &act_info)
44{
45 return kernels::ClPixelWiseMultiplicationKernel::validate(src1, src2, dst, scale, overflow_policy, rounding_policy, act_info);
46}
47
Sheri Zhangf9ab9f92021-03-16 12:09:15 +000048void ClComplexPixelWiseMultiplication::configure(const CLCompileContext &compile_context, ITensorInfo *src1, ITensorInfo *src2, ITensorInfo *dst, const ActivationLayerInfo &act_info)
49{
50 auto k = std::make_unique<kernels::ClComplexPixelWiseMultiplicationKernel>();
51 k->configure(compile_context, src1, src2, dst, act_info);
52 _kernel = std::move(k);
Sheri Zhangf9ab9f92021-03-16 12:09:15 +000053}
54
55Status ClComplexPixelWiseMultiplication::validate(const ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const ActivationLayerInfo &act_info)
56{
57 return kernels::ClComplexPixelWiseMultiplicationKernel::validate(src1, src2, dst, act_info);
58}
Sheri Zhangf9ab9f92021-03-16 12:09:15 +000059} // namespace opencl
60} // namespace arm_compute