diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-03-26 17:23:28 +0000 |
|---|---|---|
| committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-04-11 09:34:26 +0000 |
| commit | 8be9148814b88e5b0cabd5a4d2b1f4ff470a8c1c (patch) | |
| tree | 760658b8c7b8917379467bd3fc119a5502faa850 /src/core/CL/CLKernelLibrary.cpp | |
| parent | a50e702289af66944e860eafc7f3b32f6c5f30be (diff) | |
| download | ComputeLibrary-8be9148814b88e5b0cabd5a4d2b1f4ff470a8c1c.tar.gz | |
COMPMID-1959: Implements 2D FFT on OpenCL
Change-Id: I73cf3984a5463acc854c8a59dc2bd9a5234cd99c
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/936
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Diffstat (limited to 'src/core/CL/CLKernelLibrary.cpp')
| -rw-r--r-- | src/core/CL/CLKernelLibrary.cpp | 25 |
1 files changed, 24 insertions, 1 deletions
diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index 4fa8ac4142..322ff517d9 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -219,7 +219,6 @@ const std::map<std::string, std::string> CLKernelLibrary::_kernel_program_map = { "depthwise_convolution_3x3_f16", "depthwise_convolution.cl" }, { "depthwise_convolution_3x3_nhwc", "depthwise_convolution.cl" }, { "depthwise_convolution_3x3_nhwc_stride1", "depthwise_convolution.cl" }, - { "digit_reverse", "fft.cl" }, { "dwc_3x3_native_qasymm8_nchw", "depthwise_convolution_quantized.cl" }, { "dwc_3x3_native_qasymm8_dot8_nchw", "depthwise_convolution_quantized.cl" }, { "dwc_3x3_reshaped_qasymm8_nhwc", "depthwise_convolution_quantized.cl" }, @@ -261,18 +260,33 @@ const std::map<std::string, std::string> CLKernelLibrary::_kernel_program_map = { "elementwise_unary", "elementwise_unary.cl" }, { "erode", "erode.cl" }, { "fast_corners", "fast_corners.cl" }, + { "fft_digit_reverse_axis_0", "fft_digit_reverse.cl" }, + { "fft_digit_reverse_axis_1", "fft_digit_reverse.cl" }, { "fft_radix_2_first_stage_axis_0", "fft.cl" }, + { "fft_radix_2_first_stage_axis_1", "fft.cl" }, { "fft_radix_2_axis_0", "fft.cl" }, + { "fft_radix_2_axis_1", "fft.cl" }, { "fft_radix_3_first_stage_axis_0", "fft.cl" }, + { "fft_radix_3_first_stage_axis_1", "fft.cl" }, { "fft_radix_3_axis_0", "fft.cl" }, + { "fft_radix_3_axis_1", "fft.cl" }, { "fft_radix_4_first_stage_axis_0", "fft.cl" }, + { "fft_radix_4_first_stage_axis_1", "fft.cl" }, { "fft_radix_4_axis_0", "fft.cl" }, + { "fft_radix_4_axis_1", "fft.cl" }, { "fft_radix_5_first_stage_axis_0", "fft.cl" }, + { "fft_radix_5_first_stage_axis_1", "fft.cl" }, { "fft_radix_5_axis_0", "fft.cl" }, + { "fft_radix_5_axis_1", "fft.cl" }, { "fft_radix_7_first_stage_axis_0", "fft.cl" }, + { "fft_radix_7_first_stage_axis_1", "fft.cl" }, { "fft_radix_7_axis_0", "fft.cl" }, + { "fft_radix_7_axis_1", "fft.cl" }, { "fft_radix_8_first_stage_axis_0", "fft.cl" }, + { "fft_radix_8_first_stage_axis_1", "fft.cl" }, { "fft_radix_8_axis_0", "fft.cl" }, + { "fft_radix_8_axis_1", "fft.cl" }, + { "fft_scale_conj", "fft_scale.cl" }, { "fill_image_borders_constant", "fill_border.cl" }, { "fill_image_borders_replicate", "fill_border.cl" }, { "finalize", "optical_flow_pyramid_lk.cl" }, @@ -391,6 +405,7 @@ const std::map<std::string, std::string> CLKernelLibrary::_kernel_program_map = { "NV21_to_YUV444_bt709", "color_convert.cl" }, { "output_stage_quantized", "direct_convolution_1x1_3x3_5x5_quantized.cl" }, { "permute", "permute.cl" }, + { "pixelwise_mul_complex", "pixelwise_mul_float.cl" }, { "pixelwise_mul_float", "pixelwise_mul_float.cl" }, { "pixelwise_mul_int", "pixelwise_mul_int.cl" }, { "pixelwise_mul_quantized", "pixelwise_mul_int.cl" }, @@ -710,6 +725,14 @@ const std::map<std::string, std::string> CLKernelLibrary::_program_source_map = #include "./cl_kernels/fft.clembed" }, { + "fft_digit_reverse.cl", +#include "./cl_kernels/fft_digit_reverse.clembed" + }, + { + "fft_scale.cl", +#include "./cl_kernels/fft_scale.clembed" + }, + { "fill_border.cl", #include "./cl_kernels/fill_border.clembed" }, |