From 8be9148814b88e5b0cabd5a4d2b1f4ff470a8c1c Mon Sep 17 00:00:00 2001
From: Georgios Pinitas <georgios.pinitas@arm.com>
Date: Tue, 26 Mar 2019 17:23:28 +0000
Subject: 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>
---
 src/core/CL/CLKernelLibrary.cpp | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

(limited to 'src/core/CL/CLKernelLibrary.cpp')

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" },
@@ -708,6 +723,14 @@ const std::map<std::string, std::string> CLKernelLibrary::_program_source_map =
     {
         "fft.cl",
 #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",
-- 
cgit v1.2.1