mirror of
https://github.com/gnss-sdr/gnss-sdr
synced 2025-10-24 20:17:39 +00:00
Last commit from the GSoC 2013 project "Improve the acquisition sensitivity of a GNSS receiver" by Marc Molina.
Added OpenCL Acquisition blocks and tests. git-svn-id: https://svn.code.sf.net/p/gnss-sdr/code/trunk@420 64b25241-fba3-4117-9849-534c7e92360d
This commit is contained in:
@@ -25,16 +25,31 @@
|
||||
# pass_through.cc
|
||||
#)
|
||||
#else(CMAKE_CXX_COMPILER_ID MATCHES "Clang")
|
||||
set(GNSS_SPLIBS_SOURCES
|
||||
galileo_e1_signal_processing.cc
|
||||
gnss_sdr_valve.cc
|
||||
gnss_signal_processing.cc
|
||||
gps_sdr_signal_processing.cc
|
||||
nco_lib.cc
|
||||
pass_through.cc
|
||||
)
|
||||
#endif(CMAKE_CXX_COMPILER_ID MATCHES "Clang")
|
||||
|
||||
if(OPENCL_FOUND)
|
||||
set(GNSS_SPLIBS_SOURCES
|
||||
galileo_e1_signal_processing.cc
|
||||
gnss_sdr_valve.cc
|
||||
gnss_signal_processing.cc
|
||||
gps_sdr_signal_processing.cc
|
||||
nco_lib.cc
|
||||
pass_through.cc
|
||||
fft_execute.cc # Needs OpenCL
|
||||
fft_setup.cc # Needs OpenCL
|
||||
fft_kernelstring.cc # Needs OpenCL
|
||||
)
|
||||
else(OPENCL_FOUND)
|
||||
set(GNSS_SPLIBS_SOURCES
|
||||
galileo_e1_signal_processing.cc
|
||||
gnss_sdr_valve.cc
|
||||
gnss_signal_processing.cc
|
||||
gps_sdr_signal_processing.cc
|
||||
nco_lib.cc
|
||||
pass_through.cc
|
||||
)
|
||||
endif(OPENCL_FOUND)
|
||||
|
||||
include_directories(
|
||||
$(CMAKE_CURRENT_SOURCE_DIR)
|
||||
${CMAKE_SOURCE_DIR}/src/core/system_parameters
|
||||
@@ -45,5 +60,10 @@ include_directories(
|
||||
${GFlags_INCLUDE_DIRS}
|
||||
)
|
||||
|
||||
if(OPENCL_FOUND)
|
||||
include_directories( ${OPENCL_INCLUDE_DIRS} )
|
||||
set(OPT_LIBRARIES ${OPT_LIBRARIES} ${OPENCL_LIBRARIES})
|
||||
endif(OPENCL_FOUND)
|
||||
|
||||
add_library(gnss_sp_libs ${GNSS_SPLIBS_SOURCES})
|
||||
target_link_libraries(gnss_sp_libs ${GNURADIO_RUNTIME_LIBRARIES} ${GNURADIO_BLOCKS_LIBRARIES} ${GNURADIO_FFT_LIBRARIES} ${GNURADIO_FILTER_LIBRARIES} gnss_rx)
|
||||
target_link_libraries(gnss_sp_libs ${GNURADIO_RUNTIME_LIBRARIES} ${GNURADIO_BLOCKS_LIBRARIES} ${GNURADIO_FFT_LIBRARIES} ${GNURADIO_FILTER_LIBRARIES} ${OPT_LIBRARIES} gnss_rx)
|
||||
|
||||
134
src/algorithms/libs/clFFT.h
Normal file
134
src/algorithms/libs/clFFT.h
Normal file
@@ -0,0 +1,134 @@
|
||||
|
||||
//
|
||||
// File: clFFT.h
|
||||
//
|
||||
// Version: <1.0>
|
||||
//
|
||||
// Disclaimer: IMPORTANT: This Apple software is supplied to you by Apple Inc. ("Apple")
|
||||
// in consideration of your agreement to the following terms, and your use,
|
||||
// installation, modification or redistribution of this Apple software
|
||||
// constitutes acceptance of these terms. If you do not agree with these
|
||||
// terms, please do not use, install, modify or redistribute this Apple
|
||||
// software.
|
||||
//
|
||||
// In consideration of your agreement to abide by the following terms, and
|
||||
// subject to these terms, Apple grants you a personal, non - exclusive
|
||||
// license, under Apple's copyrights in this original Apple software ( the
|
||||
// "Apple Software" ), to use, reproduce, modify and redistribute the Apple
|
||||
// Software, with or without modifications, in source and / or binary forms;
|
||||
// provided that if you redistribute the Apple Software in its entirety and
|
||||
// without modifications, you must retain this notice and the following text
|
||||
// and disclaimers in all such redistributions of the Apple Software. Neither
|
||||
// the name, trademarks, service marks or logos of Apple Inc. may be used to
|
||||
// endorse or promote products derived from the Apple Software without specific
|
||||
// prior written permission from Apple. Except as expressly stated in this
|
||||
// notice, no other rights or licenses, express or implied, are granted by
|
||||
// Apple herein, including but not limited to any patent rights that may be
|
||||
// infringed by your derivative works or by other works in which the Apple
|
||||
// Software may be incorporated.
|
||||
//
|
||||
// The Apple Software is provided by Apple on an "AS IS" basis. APPLE MAKES NO
|
||||
// WARRANTIES, EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED
|
||||
// WARRANTIES OF NON - INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A
|
||||
// PARTICULAR PURPOSE, REGARDING THE APPLE SOFTWARE OR ITS USE AND OPERATION
|
||||
// ALONE OR IN COMBINATION WITH YOUR PRODUCTS.
|
||||
//
|
||||
// IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL OR
|
||||
// CONSEQUENTIAL DAMAGES ( INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION ) ARISING IN ANY WAY OUT OF THE USE, REPRODUCTION, MODIFICATION
|
||||
// AND / OR DISTRIBUTION OF THE APPLE SOFTWARE, HOWEVER CAUSED AND WHETHER
|
||||
// UNDER THEORY OF CONTRACT, TORT ( INCLUDING NEGLIGENCE ), STRICT LIABILITY OR
|
||||
// OTHERWISE, EVEN IF APPLE HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
//
|
||||
// Copyright ( C ) 2008 Apple Inc. All Rights Reserved.
|
||||
//
|
||||
////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
|
||||
#ifndef __CLFFT_H
|
||||
#define __CLFFT_H
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
#include <stdio.h>
|
||||
|
||||
#ifdef APPLE
|
||||
#include <OpenCL/cl.h>
|
||||
#else
|
||||
#include <CL/cl.h>
|
||||
#endif
|
||||
|
||||
// XForm type
|
||||
typedef enum
|
||||
{
|
||||
clFFT_Forward = -1,
|
||||
clFFT_Inverse = 1
|
||||
|
||||
}clFFT_Direction;
|
||||
|
||||
// XForm dimension
|
||||
typedef enum
|
||||
{
|
||||
clFFT_1D = 0,
|
||||
clFFT_2D = 1,
|
||||
clFFT_3D = 3
|
||||
|
||||
}clFFT_Dimension;
|
||||
|
||||
// XForm Data type
|
||||
typedef enum
|
||||
{
|
||||
clFFT_SplitComplexFormat = 0,
|
||||
clFFT_InterleavedComplexFormat = 1
|
||||
}clFFT_DataFormat;
|
||||
|
||||
typedef struct
|
||||
{
|
||||
unsigned int x;
|
||||
unsigned int y;
|
||||
unsigned int z;
|
||||
}clFFT_Dim3;
|
||||
|
||||
typedef struct
|
||||
{
|
||||
float *real;
|
||||
float *imag;
|
||||
} clFFT_SplitComplex;
|
||||
|
||||
typedef struct
|
||||
{
|
||||
float real;
|
||||
float imag;
|
||||
}clFFT_Complex;
|
||||
|
||||
typedef void* clFFT_Plan;
|
||||
|
||||
clFFT_Plan clFFT_CreatePlan( cl_context context, clFFT_Dim3 n, clFFT_Dimension dim, clFFT_DataFormat dataFormat, cl_int *error_code );
|
||||
|
||||
void clFFT_DestroyPlan( clFFT_Plan plan );
|
||||
|
||||
cl_int clFFT_ExecuteInterleaved( cl_command_queue queue, clFFT_Plan plan, cl_int batchSize, clFFT_Direction dir,
|
||||
cl_mem data_in, cl_mem data_out,
|
||||
cl_int num_events, cl_event *event_list, cl_event *event );
|
||||
|
||||
cl_int clFFT_ExecutePlannar( cl_command_queue queue, clFFT_Plan plan, cl_int batchSize, clFFT_Direction dir,
|
||||
cl_mem data_in_real, cl_mem data_in_imag, cl_mem data_out_real, cl_mem data_out_imag,
|
||||
cl_int num_events, cl_event *event_list, cl_event *event );
|
||||
|
||||
cl_int clFFT_1DTwistInterleaved(clFFT_Plan Plan, cl_command_queue queue, cl_mem array,
|
||||
size_t numRows, size_t numCols, size_t startRow, size_t rowsToProcess, clFFT_Direction dir);
|
||||
|
||||
|
||||
cl_int clFFT_1DTwistPlannar(clFFT_Plan Plan, cl_command_queue queue, cl_mem array_real, cl_mem array_imag,
|
||||
size_t numRows, size_t numCols, size_t startRow, size_t rowsToProcess, clFFT_Direction dir);
|
||||
|
||||
void clFFT_DumpPlan( clFFT_Plan plan, FILE *file);
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif
|
||||
277
src/algorithms/libs/fft_base_kernels.h
Normal file
277
src/algorithms/libs/fft_base_kernels.h
Normal file
@@ -0,0 +1,277 @@
|
||||
|
||||
//
|
||||
// File: fft_base_kernels.h
|
||||
//
|
||||
// Version: <1.0>
|
||||
//
|
||||
// Disclaimer: IMPORTANT: This Apple software is supplied to you by Apple Inc. ("Apple")
|
||||
// in consideration of your agreement to the following terms, and your use,
|
||||
// installation, modification or redistribution of this Apple software
|
||||
// constitutes acceptance of these terms. If you do not agree with these
|
||||
// terms, please do not use, install, modify or redistribute this Apple
|
||||
// software.
|
||||
//
|
||||
// In consideration of your agreement to abide by the following terms, and
|
||||
// subject to these terms, Apple grants you a personal, non - exclusive
|
||||
// license, under Apple's copyrights in this original Apple software ( the
|
||||
// "Apple Software" ), to use, reproduce, modify and redistribute the Apple
|
||||
// Software, with or without modifications, in source and / or binary forms;
|
||||
// provided that if you redistribute the Apple Software in its entirety and
|
||||
// without modifications, you must retain this notice and the following text
|
||||
// and disclaimers in all such redistributions of the Apple Software. Neither
|
||||
// the name, trademarks, service marks or logos of Apple Inc. may be used to
|
||||
// endorse or promote products derived from the Apple Software without specific
|
||||
// prior written permission from Apple. Except as expressly stated in this
|
||||
// notice, no other rights or licenses, express or implied, are granted by
|
||||
// Apple herein, including but not limited to any patent rights that may be
|
||||
// infringed by your derivative works or by other works in which the Apple
|
||||
// Software may be incorporated.
|
||||
//
|
||||
// The Apple Software is provided by Apple on an "AS IS" basis. APPLE MAKES NO
|
||||
// WARRANTIES, EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED
|
||||
// WARRANTIES OF NON - INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A
|
||||
// PARTICULAR PURPOSE, REGARDING THE APPLE SOFTWARE OR ITS USE AND OPERATION
|
||||
// ALONE OR IN COMBINATION WITH YOUR PRODUCTS.
|
||||
//
|
||||
// IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL OR
|
||||
// CONSEQUENTIAL DAMAGES ( INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION ) ARISING IN ANY WAY OUT OF THE USE, REPRODUCTION, MODIFICATION
|
||||
// AND / OR DISTRIBUTION OF THE APPLE SOFTWARE, HOWEVER CAUSED AND WHETHER
|
||||
// UNDER THEORY OF CONTRACT, TORT ( INCLUDING NEGLIGENCE ), STRICT LIABILITY OR
|
||||
// OTHERWISE, EVEN IF APPLE HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
//
|
||||
// Copyright ( C ) 2008 Apple Inc. All Rights Reserved.
|
||||
//
|
||||
////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
|
||||
#ifndef __CL_FFT_BASE_KERNELS_
|
||||
#define __CL_FFT_BASE_KERNELS_
|
||||
|
||||
#include <string>
|
||||
|
||||
using namespace std;
|
||||
|
||||
static string baseKernels = string(
|
||||
"#ifndef M_PI\n"
|
||||
"#define M_PI 0x1.921fb54442d18p+1\n"
|
||||
"#endif\n"
|
||||
"#define complexMul(a,b) ((float2)(mad(-(a).y, (b).y, (a).x * (b).x), mad((a).y, (b).x, (a).x * (b).y)))\n"
|
||||
"#define conj(a) ((float2)((a).x, -(a).y))\n"
|
||||
"#define conjTransp(a) ((float2)(-(a).y, (a).x))\n"
|
||||
"\n"
|
||||
"#define fftKernel2(a,dir) \\\n"
|
||||
"{ \\\n"
|
||||
" float2 c = (a)[0]; \\\n"
|
||||
" (a)[0] = c + (a)[1]; \\\n"
|
||||
" (a)[1] = c - (a)[1]; \\\n"
|
||||
"}\n"
|
||||
"\n"
|
||||
"#define fftKernel2S(d1,d2,dir) \\\n"
|
||||
"{ \\\n"
|
||||
" float2 c = (d1); \\\n"
|
||||
" (d1) = c + (d2); \\\n"
|
||||
" (d2) = c - (d2); \\\n"
|
||||
"}\n"
|
||||
"\n"
|
||||
"#define fftKernel4(a,dir) \\\n"
|
||||
"{ \\\n"
|
||||
" fftKernel2S((a)[0], (a)[2], dir); \\\n"
|
||||
" fftKernel2S((a)[1], (a)[3], dir); \\\n"
|
||||
" fftKernel2S((a)[0], (a)[1], dir); \\\n"
|
||||
" (a)[3] = (float2)(dir)*(conjTransp((a)[3])); \\\n"
|
||||
" fftKernel2S((a)[2], (a)[3], dir); \\\n"
|
||||
" float2 c = (a)[1]; \\\n"
|
||||
" (a)[1] = (a)[2]; \\\n"
|
||||
" (a)[2] = c; \\\n"
|
||||
"}\n"
|
||||
"\n"
|
||||
"#define fftKernel4s(a0,a1,a2,a3,dir) \\\n"
|
||||
"{ \\\n"
|
||||
" fftKernel2S((a0), (a2), dir); \\\n"
|
||||
" fftKernel2S((a1), (a3), dir); \\\n"
|
||||
" fftKernel2S((a0), (a1), dir); \\\n"
|
||||
" (a3) = (float2)(dir)*(conjTransp((a3))); \\\n"
|
||||
" fftKernel2S((a2), (a3), dir); \\\n"
|
||||
" float2 c = (a1); \\\n"
|
||||
" (a1) = (a2); \\\n"
|
||||
" (a2) = c; \\\n"
|
||||
"}\n"
|
||||
"\n"
|
||||
"#define bitreverse8(a) \\\n"
|
||||
"{ \\\n"
|
||||
" float2 c; \\\n"
|
||||
" c = (a)[1]; \\\n"
|
||||
" (a)[1] = (a)[4]; \\\n"
|
||||
" (a)[4] = c; \\\n"
|
||||
" c = (a)[3]; \\\n"
|
||||
" (a)[3] = (a)[6]; \\\n"
|
||||
" (a)[6] = c; \\\n"
|
||||
"}\n"
|
||||
"\n"
|
||||
"#define fftKernel8(a,dir) \\\n"
|
||||
"{ \\\n"
|
||||
" const float2 w1 = (float2)(0x1.6a09e6p-1f, dir*0x1.6a09e6p-1f); \\\n"
|
||||
" const float2 w3 = (float2)(-0x1.6a09e6p-1f, dir*0x1.6a09e6p-1f); \\\n"
|
||||
" float2 c; \\\n"
|
||||
" fftKernel2S((a)[0], (a)[4], dir); \\\n"
|
||||
" fftKernel2S((a)[1], (a)[5], dir); \\\n"
|
||||
" fftKernel2S((a)[2], (a)[6], dir); \\\n"
|
||||
" fftKernel2S((a)[3], (a)[7], dir); \\\n"
|
||||
" (a)[5] = complexMul(w1, (a)[5]); \\\n"
|
||||
" (a)[6] = (float2)(dir)*(conjTransp((a)[6])); \\\n"
|
||||
" (a)[7] = complexMul(w3, (a)[7]); \\\n"
|
||||
" fftKernel2S((a)[0], (a)[2], dir); \\\n"
|
||||
" fftKernel2S((a)[1], (a)[3], dir); \\\n"
|
||||
" fftKernel2S((a)[4], (a)[6], dir); \\\n"
|
||||
" fftKernel2S((a)[5], (a)[7], dir); \\\n"
|
||||
" (a)[3] = (float2)(dir)*(conjTransp((a)[3])); \\\n"
|
||||
" (a)[7] = (float2)(dir)*(conjTransp((a)[7])); \\\n"
|
||||
" fftKernel2S((a)[0], (a)[1], dir); \\\n"
|
||||
" fftKernel2S((a)[2], (a)[3], dir); \\\n"
|
||||
" fftKernel2S((a)[4], (a)[5], dir); \\\n"
|
||||
" fftKernel2S((a)[6], (a)[7], dir); \\\n"
|
||||
" bitreverse8((a)); \\\n"
|
||||
"}\n"
|
||||
"\n"
|
||||
"#define bitreverse4x4(a) \\\n"
|
||||
"{ \\\n"
|
||||
" float2 c; \\\n"
|
||||
" c = (a)[1]; (a)[1] = (a)[4]; (a)[4] = c; \\\n"
|
||||
" c = (a)[2]; (a)[2] = (a)[8]; (a)[8] = c; \\\n"
|
||||
" c = (a)[3]; (a)[3] = (a)[12]; (a)[12] = c; \\\n"
|
||||
" c = (a)[6]; (a)[6] = (a)[9]; (a)[9] = c; \\\n"
|
||||
" c = (a)[7]; (a)[7] = (a)[13]; (a)[13] = c; \\\n"
|
||||
" c = (a)[11]; (a)[11] = (a)[14]; (a)[14] = c; \\\n"
|
||||
"}\n"
|
||||
"\n"
|
||||
"#define fftKernel16(a,dir) \\\n"
|
||||
"{ \\\n"
|
||||
" const float w0 = 0x1.d906bcp-1f; \\\n"
|
||||
" const float w1 = 0x1.87de2ap-2f; \\\n"
|
||||
" const float w2 = 0x1.6a09e6p-1f; \\\n"
|
||||
" fftKernel4s((a)[0], (a)[4], (a)[8], (a)[12], dir); \\\n"
|
||||
" fftKernel4s((a)[1], (a)[5], (a)[9], (a)[13], dir); \\\n"
|
||||
" fftKernel4s((a)[2], (a)[6], (a)[10], (a)[14], dir); \\\n"
|
||||
" fftKernel4s((a)[3], (a)[7], (a)[11], (a)[15], dir); \\\n"
|
||||
" (a)[5] = complexMul((a)[5], (float2)(w0, dir*w1)); \\\n"
|
||||
" (a)[6] = complexMul((a)[6], (float2)(w2, dir*w2)); \\\n"
|
||||
" (a)[7] = complexMul((a)[7], (float2)(w1, dir*w0)); \\\n"
|
||||
" (a)[9] = complexMul((a)[9], (float2)(w2, dir*w2)); \\\n"
|
||||
" (a)[10] = (float2)(dir)*(conjTransp((a)[10])); \\\n"
|
||||
" (a)[11] = complexMul((a)[11], (float2)(-w2, dir*w2)); \\\n"
|
||||
" (a)[13] = complexMul((a)[13], (float2)(w1, dir*w0)); \\\n"
|
||||
" (a)[14] = complexMul((a)[14], (float2)(-w2, dir*w2)); \\\n"
|
||||
" (a)[15] = complexMul((a)[15], (float2)(-w0, dir*-w1)); \\\n"
|
||||
" fftKernel4((a), dir); \\\n"
|
||||
" fftKernel4((a) + 4, dir); \\\n"
|
||||
" fftKernel4((a) + 8, dir); \\\n"
|
||||
" fftKernel4((a) + 12, dir); \\\n"
|
||||
" bitreverse4x4((a)); \\\n"
|
||||
"}\n"
|
||||
"\n"
|
||||
"#define bitreverse32(a) \\\n"
|
||||
"{ \\\n"
|
||||
" float2 c1, c2; \\\n"
|
||||
" c1 = (a)[2]; (a)[2] = (a)[1]; c2 = (a)[4]; (a)[4] = c1; c1 = (a)[8]; (a)[8] = c2; c2 = (a)[16]; (a)[16] = c1; (a)[1] = c2; \\\n"
|
||||
" c1 = (a)[6]; (a)[6] = (a)[3]; c2 = (a)[12]; (a)[12] = c1; c1 = (a)[24]; (a)[24] = c2; c2 = (a)[17]; (a)[17] = c1; (a)[3] = c2; \\\n"
|
||||
" c1 = (a)[10]; (a)[10] = (a)[5]; c2 = (a)[20]; (a)[20] = c1; c1 = (a)[9]; (a)[9] = c2; c2 = (a)[18]; (a)[18] = c1; (a)[5] = c2; \\\n"
|
||||
" c1 = (a)[14]; (a)[14] = (a)[7]; c2 = (a)[28]; (a)[28] = c1; c1 = (a)[25]; (a)[25] = c2; c2 = (a)[19]; (a)[19] = c1; (a)[7] = c2; \\\n"
|
||||
" c1 = (a)[22]; (a)[22] = (a)[11]; c2 = (a)[13]; (a)[13] = c1; c1 = (a)[26]; (a)[26] = c2; c2 = (a)[21]; (a)[21] = c1; (a)[11] = c2; \\\n"
|
||||
" c1 = (a)[30]; (a)[30] = (a)[15]; c2 = (a)[29]; (a)[29] = c1; c1 = (a)[27]; (a)[27] = c2; c2 = (a)[23]; (a)[23] = c1; (a)[15] = c2; \\\n"
|
||||
"}\n"
|
||||
"\n"
|
||||
"#define fftKernel32(a,dir) \\\n"
|
||||
"{ \\\n"
|
||||
" fftKernel2S((a)[0], (a)[16], dir); \\\n"
|
||||
" fftKernel2S((a)[1], (a)[17], dir); \\\n"
|
||||
" fftKernel2S((a)[2], (a)[18], dir); \\\n"
|
||||
" fftKernel2S((a)[3], (a)[19], dir); \\\n"
|
||||
" fftKernel2S((a)[4], (a)[20], dir); \\\n"
|
||||
" fftKernel2S((a)[5], (a)[21], dir); \\\n"
|
||||
" fftKernel2S((a)[6], (a)[22], dir); \\\n"
|
||||
" fftKernel2S((a)[7], (a)[23], dir); \\\n"
|
||||
" fftKernel2S((a)[8], (a)[24], dir); \\\n"
|
||||
" fftKernel2S((a)[9], (a)[25], dir); \\\n"
|
||||
" fftKernel2S((a)[10], (a)[26], dir); \\\n"
|
||||
" fftKernel2S((a)[11], (a)[27], dir); \\\n"
|
||||
" fftKernel2S((a)[12], (a)[28], dir); \\\n"
|
||||
" fftKernel2S((a)[13], (a)[29], dir); \\\n"
|
||||
" fftKernel2S((a)[14], (a)[30], dir); \\\n"
|
||||
" fftKernel2S((a)[15], (a)[31], dir); \\\n"
|
||||
" (a)[17] = complexMul((a)[17], (float2)(0x1.f6297cp-1f, dir*0x1.8f8b84p-3f)); \\\n"
|
||||
" (a)[18] = complexMul((a)[18], (float2)(0x1.d906bcp-1f, dir*0x1.87de2ap-2f)); \\\n"
|
||||
" (a)[19] = complexMul((a)[19], (float2)(0x1.a9b662p-1f, dir*0x1.1c73b4p-1f)); \\\n"
|
||||
" (a)[20] = complexMul((a)[20], (float2)(0x1.6a09e6p-1f, dir*0x1.6a09e6p-1f)); \\\n"
|
||||
" (a)[21] = complexMul((a)[21], (float2)(0x1.1c73b4p-1f, dir*0x1.a9b662p-1f)); \\\n"
|
||||
" (a)[22] = complexMul((a)[22], (float2)(0x1.87de2ap-2f, dir*0x1.d906bcp-1f)); \\\n"
|
||||
" (a)[23] = complexMul((a)[23], (float2)(0x1.8f8b84p-3f, dir*0x1.f6297cp-1f)); \\\n"
|
||||
" (a)[24] = complexMul((a)[24], (float2)(0x0p+0f, dir*0x1p+0f)); \\\n"
|
||||
" (a)[25] = complexMul((a)[25], (float2)(-0x1.8f8b84p-3f, dir*0x1.f6297cp-1f)); \\\n"
|
||||
" (a)[26] = complexMul((a)[26], (float2)(-0x1.87de2ap-2f, dir*0x1.d906bcp-1f)); \\\n"
|
||||
" (a)[27] = complexMul((a)[27], (float2)(-0x1.1c73b4p-1f, dir*0x1.a9b662p-1f)); \\\n"
|
||||
" (a)[28] = complexMul((a)[28], (float2)(-0x1.6a09e6p-1f, dir*0x1.6a09e6p-1f)); \\\n"
|
||||
" (a)[29] = complexMul((a)[29], (float2)(-0x1.a9b662p-1f, dir*0x1.1c73b4p-1f)); \\\n"
|
||||
" (a)[30] = complexMul((a)[30], (float2)(-0x1.d906bcp-1f, dir*0x1.87de2ap-2f)); \\\n"
|
||||
" (a)[31] = complexMul((a)[31], (float2)(-0x1.f6297cp-1f, dir*0x1.8f8b84p-3f)); \\\n"
|
||||
" fftKernel16((a), dir); \\\n"
|
||||
" fftKernel16((a) + 16, dir); \\\n"
|
||||
" bitreverse32((a)); \\\n"
|
||||
"}\n\n"
|
||||
);
|
||||
|
||||
static string twistKernelInterleaved = string(
|
||||
"__kernel void \\\n"
|
||||
"clFFT_1DTwistInterleaved(__global float2 *in, unsigned int startRow, unsigned int numCols, unsigned int N, unsigned int numRowsToProcess, int dir) \\\n"
|
||||
"{ \\\n"
|
||||
" float2 a, w; \\\n"
|
||||
" float ang; \\\n"
|
||||
" unsigned int j; \\\n"
|
||||
" unsigned int i = get_global_id(0); \\\n"
|
||||
" unsigned int startIndex = i; \\\n"
|
||||
" \\\n"
|
||||
" if(i < numCols) \\\n"
|
||||
" { \\\n"
|
||||
" for(j = 0; j < numRowsToProcess; j++) \\\n"
|
||||
" { \\\n"
|
||||
" a = in[startIndex]; \\\n"
|
||||
" ang = 2.0f * M_PI * dir * i * (startRow + j) / N; \\\n"
|
||||
" w = (float2)(native_cos(ang), native_sin(ang)); \\\n"
|
||||
" a = complexMul(a, w); \\\n"
|
||||
" in[startIndex] = a; \\\n"
|
||||
" startIndex += numCols; \\\n"
|
||||
" } \\\n"
|
||||
" } \\\n"
|
||||
"} \\\n"
|
||||
);
|
||||
|
||||
static string twistKernelPlannar = string(
|
||||
"__kernel void \\\n"
|
||||
"clFFT_1DTwistSplit(__global float *in_real, __global float *in_imag , unsigned int startRow, unsigned int numCols, unsigned int N, unsigned int numRowsToProcess, int dir) \\\n"
|
||||
"{ \\\n"
|
||||
" float2 a, w; \\\n"
|
||||
" float ang; \\\n"
|
||||
" unsigned int j; \\\n"
|
||||
" unsigned int i = get_global_id(0); \\\n"
|
||||
" unsigned int startIndex = i; \\\n"
|
||||
" \\\n"
|
||||
" if(i < numCols) \\\n"
|
||||
" { \\\n"
|
||||
" for(j = 0; j < numRowsToProcess; j++) \\\n"
|
||||
" { \\\n"
|
||||
" a = (float2)(in_real[startIndex], in_imag[startIndex]); \\\n"
|
||||
" ang = 2.0f * M_PI * dir * i * (startRow + j) / N; \\\n"
|
||||
" w = (float2)(native_cos(ang), native_sin(ang)); \\\n"
|
||||
" a = complexMul(a, w); \\\n"
|
||||
" in_real[startIndex] = a.x; \\\n"
|
||||
" in_imag[startIndex] = a.y; \\\n"
|
||||
" startIndex += numCols; \\\n"
|
||||
" } \\\n"
|
||||
" } \\\n"
|
||||
"} \\\n"
|
||||
);
|
||||
|
||||
|
||||
|
||||
#endif
|
||||
405
src/algorithms/libs/fft_execute.cc
Normal file
405
src/algorithms/libs/fft_execute.cc
Normal file
@@ -0,0 +1,405 @@
|
||||
|
||||
//
|
||||
// File: fft_execute.cpp
|
||||
//
|
||||
// Version: <1.0>
|
||||
//
|
||||
// Disclaimer: IMPORTANT: This Apple software is supplied to you by Apple Inc. ("Apple")
|
||||
// in consideration of your agreement to the following terms, and your use,
|
||||
// installation, modification or redistribution of this Apple software
|
||||
// constitutes acceptance of these terms. If you do not agree with these
|
||||
// terms, please do not use, install, modify or redistribute this Apple
|
||||
// software.¬
|
||||
//
|
||||
// In consideration of your agreement to abide by the following terms, and
|
||||
// subject to these terms, Apple grants you a personal, non - exclusive
|
||||
// license, under Apple's copyrights in this original Apple software ( the
|
||||
// "Apple Software" ), to use, reproduce, modify and redistribute the Apple
|
||||
// Software, with or without modifications, in source and / or binary forms;
|
||||
// provided that if you redistribute the Apple Software in its entirety and
|
||||
// without modifications, you must retain this notice and the following text
|
||||
// and disclaimers in all such redistributions of the Apple Software. Neither
|
||||
// the name, trademarks, service marks or logos of Apple Inc. may be used to
|
||||
// endorse or promote products derived from the Apple Software without specific
|
||||
// prior written permission from Apple. Except as expressly stated in this
|
||||
// notice, no other rights or licenses, express or implied, are granted by
|
||||
// Apple herein, including but not limited to any patent rights that may be
|
||||
// infringed by your derivative works or by other works in which the Apple
|
||||
// Software may be incorporated.
|
||||
//
|
||||
// The Apple Software is provided by Apple on an "AS IS" basis. APPLE MAKES NO
|
||||
// WARRANTIES, EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED
|
||||
// WARRANTIES OF NON - INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A
|
||||
// PARTICULAR PURPOSE, REGARDING THE APPLE SOFTWARE OR ITS USE AND OPERATION
|
||||
// ALONE OR IN COMBINATION WITH YOUR PRODUCTS.
|
||||
//
|
||||
// IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL OR
|
||||
// CONSEQUENTIAL DAMAGES ( INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION ) ARISING IN ANY WAY OUT OF THE USE, REPRODUCTION, MODIFICATION
|
||||
// AND / OR DISTRIBUTION OF THE APPLE SOFTWARE, HOWEVER CAUSED AND WHETHER
|
||||
// UNDER THEORY OF CONTRACT, TORT ( INCLUDING NEGLIGENCE ), STRICT LIABILITY OR
|
||||
// OTHERWISE, EVEN IF APPLE HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
//
|
||||
// Copyright ( C ) 2008 Apple Inc. All Rights Reserved.
|
||||
//
|
||||
////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
|
||||
#include "fft_internal.h"
|
||||
#include "clFFT.h"
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <math.h>
|
||||
|
||||
#define max(a,b) (((a)>(b)) ? (a) : (b))
|
||||
#define min(a,b) (((a)<(b)) ? (a) : (b))
|
||||
|
||||
static cl_int
|
||||
allocateTemporaryBufferInterleaved(cl_fft_plan *plan, cl_uint batchSize)
|
||||
{
|
||||
cl_int err = CL_SUCCESS;
|
||||
if(plan->temp_buffer_needed && plan->last_batch_size != batchSize)
|
||||
{
|
||||
plan->last_batch_size = batchSize;
|
||||
size_t tmpLength = plan->n.x * plan->n.y * plan->n.z * batchSize * 2 * sizeof(cl_float);
|
||||
|
||||
if(plan->tempmemobj)
|
||||
clReleaseMemObject(plan->tempmemobj);
|
||||
|
||||
plan->tempmemobj = clCreateBuffer(plan->context, CL_MEM_READ_WRITE, tmpLength, NULL, &err);
|
||||
}
|
||||
return err;
|
||||
}
|
||||
|
||||
static cl_int
|
||||
allocateTemporaryBufferPlannar(cl_fft_plan *plan, cl_uint batchSize)
|
||||
{
|
||||
cl_int err = CL_SUCCESS;
|
||||
cl_int terr;
|
||||
if(plan->temp_buffer_needed && plan->last_batch_size != batchSize)
|
||||
{
|
||||
plan->last_batch_size = batchSize;
|
||||
size_t tmpLength = plan->n.x * plan->n.y * plan->n.z * batchSize * sizeof(cl_float);
|
||||
|
||||
if(plan->tempmemobj_real)
|
||||
clReleaseMemObject(plan->tempmemobj_real);
|
||||
|
||||
if(plan->tempmemobj_imag)
|
||||
clReleaseMemObject(plan->tempmemobj_imag);
|
||||
|
||||
plan->tempmemobj_real = clCreateBuffer(plan->context, CL_MEM_READ_WRITE, tmpLength, NULL, &err);
|
||||
plan->tempmemobj_imag = clCreateBuffer(plan->context, CL_MEM_READ_WRITE, tmpLength, NULL, &terr);
|
||||
err |= terr;
|
||||
}
|
||||
return err;
|
||||
}
|
||||
|
||||
void
|
||||
getKernelWorkDimensions(cl_fft_plan *plan, cl_fft_kernel_info *kernelInfo, cl_int *batchSize, size_t *gWorkItems, size_t *lWorkItems)
|
||||
{
|
||||
*lWorkItems = kernelInfo->num_workitems_per_workgroup;
|
||||
int numWorkGroups = kernelInfo->num_workgroups;
|
||||
int numXFormsPerWG = kernelInfo->num_xforms_per_workgroup;
|
||||
|
||||
switch(kernelInfo->dir)
|
||||
{
|
||||
case cl_fft_kernel_x:
|
||||
*batchSize *= (plan->n.y * plan->n.z);
|
||||
numWorkGroups = (*batchSize % numXFormsPerWG) ? (*batchSize/numXFormsPerWG + 1) : (*batchSize/numXFormsPerWG);
|
||||
numWorkGroups *= kernelInfo->num_workgroups;
|
||||
break;
|
||||
case cl_fft_kernel_y:
|
||||
*batchSize *= plan->n.z;
|
||||
numWorkGroups *= *batchSize;
|
||||
break;
|
||||
case cl_fft_kernel_z:
|
||||
numWorkGroups *= *batchSize;
|
||||
break;
|
||||
}
|
||||
|
||||
*gWorkItems = numWorkGroups * *lWorkItems;
|
||||
}
|
||||
|
||||
cl_int
|
||||
clFFT_ExecuteInterleaved( cl_command_queue queue, clFFT_Plan Plan, cl_int batchSize, clFFT_Direction dir,
|
||||
cl_mem data_in, cl_mem data_out,
|
||||
cl_int num_events, cl_event *event_list, cl_event *event )
|
||||
{
|
||||
int s;
|
||||
cl_fft_plan *plan = (cl_fft_plan *) Plan;
|
||||
if(plan->format != clFFT_InterleavedComplexFormat)
|
||||
return CL_INVALID_VALUE;
|
||||
|
||||
cl_int err;
|
||||
size_t gWorkItems, lWorkItems;
|
||||
int inPlaceDone;
|
||||
|
||||
cl_int isInPlace = data_in == data_out ? 1 : 0;
|
||||
|
||||
if((err = allocateTemporaryBufferInterleaved(plan, batchSize)) != CL_SUCCESS)
|
||||
return err;
|
||||
|
||||
cl_mem memObj[3];
|
||||
memObj[0] = data_in;
|
||||
memObj[1] = data_out;
|
||||
memObj[2] = plan->tempmemobj;
|
||||
cl_fft_kernel_info *kernelInfo = plan->kernel_info;
|
||||
int numKernels = plan->num_kernels;
|
||||
|
||||
int numKernelsOdd = numKernels & 1;
|
||||
int currRead = 0;
|
||||
int currWrite = 1;
|
||||
|
||||
// at least one external dram shuffle (transpose) required
|
||||
if(plan->temp_buffer_needed)
|
||||
{
|
||||
// in-place transform
|
||||
if(isInPlace)
|
||||
{
|
||||
inPlaceDone = 0;
|
||||
currRead = 1;
|
||||
currWrite = 2;
|
||||
}
|
||||
else
|
||||
{
|
||||
currWrite = (numKernels & 1) ? 1 : 2;
|
||||
}
|
||||
|
||||
while(kernelInfo)
|
||||
{
|
||||
if( isInPlace && numKernelsOdd && !inPlaceDone && kernelInfo->in_place_possible)
|
||||
{
|
||||
currWrite = currRead;
|
||||
inPlaceDone = 1;
|
||||
}
|
||||
|
||||
s = batchSize;
|
||||
getKernelWorkDimensions(plan, kernelInfo, &s, &gWorkItems, &lWorkItems);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 0, sizeof(cl_mem), &memObj[currRead]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 1, sizeof(cl_mem), &memObj[currWrite]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 2, sizeof(cl_int), &dir);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 3, sizeof(cl_int), &s);
|
||||
|
||||
err |= clEnqueueNDRangeKernel(queue, kernelInfo->kernel, 1, NULL, &gWorkItems, &lWorkItems, 0, NULL, NULL);
|
||||
if(err)
|
||||
return err;
|
||||
|
||||
currRead = (currWrite == 1) ? 1 : 2;
|
||||
currWrite = (currWrite == 1) ? 2 : 1;
|
||||
|
||||
kernelInfo = kernelInfo->next;
|
||||
}
|
||||
}
|
||||
// no dram shuffle (transpose required) transform
|
||||
// all kernels can execute in-place.
|
||||
else {
|
||||
|
||||
while(kernelInfo)
|
||||
{
|
||||
s = batchSize;
|
||||
getKernelWorkDimensions(plan, kernelInfo, &s, &gWorkItems, &lWorkItems);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 0, sizeof(cl_mem), &memObj[currRead]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 1, sizeof(cl_mem), &memObj[currWrite]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 2, sizeof(cl_int), &dir);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 3, sizeof(cl_int), &s);
|
||||
|
||||
err |= clEnqueueNDRangeKernel(queue, kernelInfo->kernel, 1, NULL, &gWorkItems, &lWorkItems, 0, NULL, NULL);
|
||||
if(err)
|
||||
return err;
|
||||
|
||||
currRead = 1;
|
||||
currWrite = 1;
|
||||
|
||||
kernelInfo = kernelInfo->next;
|
||||
}
|
||||
}
|
||||
|
||||
return err;
|
||||
}
|
||||
|
||||
cl_int
|
||||
clFFT_ExecutePlannar( cl_command_queue queue, clFFT_Plan Plan, cl_int batchSize, clFFT_Direction dir,
|
||||
cl_mem data_in_real, cl_mem data_in_imag, cl_mem data_out_real, cl_mem data_out_imag,
|
||||
cl_int num_events, cl_event *event_list, cl_event *event)
|
||||
{
|
||||
int s;
|
||||
cl_fft_plan *plan = (cl_fft_plan *) Plan;
|
||||
|
||||
if(plan->format != clFFT_SplitComplexFormat)
|
||||
return CL_INVALID_VALUE;
|
||||
|
||||
cl_int err;
|
||||
size_t gWorkItems, lWorkItems;
|
||||
int inPlaceDone;
|
||||
|
||||
cl_int isInPlace = ((data_in_real == data_out_real) && (data_in_imag == data_out_imag)) ? 1 : 0;
|
||||
|
||||
if((err = allocateTemporaryBufferPlannar(plan, batchSize)) != CL_SUCCESS)
|
||||
return err;
|
||||
|
||||
cl_mem memObj_real[3];
|
||||
cl_mem memObj_imag[3];
|
||||
memObj_real[0] = data_in_real;
|
||||
memObj_real[1] = data_out_real;
|
||||
memObj_real[2] = plan->tempmemobj_real;
|
||||
memObj_imag[0] = data_in_imag;
|
||||
memObj_imag[1] = data_out_imag;
|
||||
memObj_imag[2] = plan->tempmemobj_imag;
|
||||
|
||||
cl_fft_kernel_info *kernelInfo = plan->kernel_info;
|
||||
int numKernels = plan->num_kernels;
|
||||
|
||||
int numKernelsOdd = numKernels & 1;
|
||||
int currRead = 0;
|
||||
int currWrite = 1;
|
||||
|
||||
// at least one external dram shuffle (transpose) required
|
||||
if(plan->temp_buffer_needed)
|
||||
{
|
||||
// in-place transform
|
||||
if(isInPlace)
|
||||
{
|
||||
inPlaceDone = 0;
|
||||
currRead = 1;
|
||||
currWrite = 2;
|
||||
}
|
||||
else
|
||||
{
|
||||
currWrite = (numKernels & 1) ? 1 : 2;
|
||||
}
|
||||
|
||||
while(kernelInfo)
|
||||
{
|
||||
if( isInPlace && numKernelsOdd && !inPlaceDone && kernelInfo->in_place_possible)
|
||||
{
|
||||
currWrite = currRead;
|
||||
inPlaceDone = 1;
|
||||
}
|
||||
|
||||
s = batchSize;
|
||||
getKernelWorkDimensions(plan, kernelInfo, &s, &gWorkItems, &lWorkItems);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 0, sizeof(cl_mem), &memObj_real[currRead]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 1, sizeof(cl_mem), &memObj_imag[currRead]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 2, sizeof(cl_mem), &memObj_real[currWrite]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 3, sizeof(cl_mem), &memObj_imag[currWrite]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 4, sizeof(cl_int), &dir);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 5, sizeof(cl_int), &s);
|
||||
|
||||
err |= clEnqueueNDRangeKernel(queue, kernelInfo->kernel, 1, NULL, &gWorkItems, &lWorkItems, 0, NULL, NULL);
|
||||
if(err)
|
||||
return err;
|
||||
|
||||
currRead = (currWrite == 1) ? 1 : 2;
|
||||
currWrite = (currWrite == 1) ? 2 : 1;
|
||||
|
||||
kernelInfo = kernelInfo->next;
|
||||
}
|
||||
}
|
||||
// no dram shuffle (transpose required) transform
|
||||
else {
|
||||
|
||||
while(kernelInfo)
|
||||
{
|
||||
s = batchSize;
|
||||
getKernelWorkDimensions(plan, kernelInfo, &s, &gWorkItems, &lWorkItems);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 0, sizeof(cl_mem), &memObj_real[currRead]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 1, sizeof(cl_mem), &memObj_imag[currRead]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 2, sizeof(cl_mem), &memObj_real[currWrite]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 3, sizeof(cl_mem), &memObj_imag[currWrite]);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 4, sizeof(cl_int), &dir);
|
||||
err |= clSetKernelArg(kernelInfo->kernel, 5, sizeof(cl_int), &s);
|
||||
|
||||
err |= clEnqueueNDRangeKernel(queue, kernelInfo->kernel, 1, NULL, &gWorkItems, &lWorkItems, 0, NULL, NULL);
|
||||
if(err)
|
||||
return err;
|
||||
|
||||
currRead = 1;
|
||||
currWrite = 1;
|
||||
|
||||
kernelInfo = kernelInfo->next;
|
||||
}
|
||||
}
|
||||
|
||||
return err;
|
||||
}
|
||||
|
||||
cl_int
|
||||
clFFT_1DTwistInterleaved(clFFT_Plan Plan, cl_command_queue queue, cl_mem array,
|
||||
unsigned numRows, unsigned numCols, unsigned startRow, unsigned rowsToProcess, clFFT_Direction dir)
|
||||
{
|
||||
cl_fft_plan *plan = (cl_fft_plan *) Plan;
|
||||
|
||||
unsigned int N = numRows*numCols;
|
||||
unsigned int nCols = numCols;
|
||||
unsigned int sRow = startRow;
|
||||
unsigned int rToProcess = rowsToProcess;
|
||||
int d = dir;
|
||||
int err = 0;
|
||||
|
||||
cl_device_id device_id;
|
||||
err = clGetCommandQueueInfo(queue, CL_QUEUE_DEVICE, sizeof(cl_device_id), &device_id, NULL);
|
||||
if(err)
|
||||
return err;
|
||||
|
||||
size_t gSize;
|
||||
err = clGetKernelWorkGroupInfo(plan->twist_kernel, device_id, CL_KERNEL_WORK_GROUP_SIZE, sizeof(size_t), &gSize, NULL);
|
||||
if(err)
|
||||
return err;
|
||||
|
||||
gSize = min(128, gSize);
|
||||
size_t numGlobalThreads[1] = { max(numCols / gSize, 1)*gSize };
|
||||
size_t numLocalThreads[1] = { gSize };
|
||||
|
||||
err |= clSetKernelArg(plan->twist_kernel, 0, sizeof(cl_mem), &array);
|
||||
err |= clSetKernelArg(plan->twist_kernel, 1, sizeof(unsigned int), &sRow);
|
||||
err |= clSetKernelArg(plan->twist_kernel, 2, sizeof(unsigned int), &nCols);
|
||||
err |= clSetKernelArg(plan->twist_kernel, 3, sizeof(unsigned int), &N);
|
||||
err |= clSetKernelArg(plan->twist_kernel, 4, sizeof(unsigned int), &rToProcess);
|
||||
err |= clSetKernelArg(plan->twist_kernel, 5, sizeof(int), &d);
|
||||
|
||||
err |= clEnqueueNDRangeKernel(queue, plan->twist_kernel, 1, NULL, numGlobalThreads, numLocalThreads, 0, NULL, NULL);
|
||||
|
||||
return err;
|
||||
}
|
||||
|
||||
cl_int
|
||||
clFFT_1DTwistPlannar(clFFT_Plan Plan, cl_command_queue queue, cl_mem array_real, cl_mem array_imag,
|
||||
unsigned numRows, unsigned numCols, unsigned startRow, unsigned rowsToProcess, clFFT_Direction dir)
|
||||
{
|
||||
cl_fft_plan *plan = (cl_fft_plan *) Plan;
|
||||
|
||||
unsigned int N = numRows*numCols;
|
||||
unsigned int nCols = numCols;
|
||||
unsigned int sRow = startRow;
|
||||
unsigned int rToProcess = rowsToProcess;
|
||||
int d = dir;
|
||||
int err = 0;
|
||||
|
||||
cl_device_id device_id;
|
||||
err = clGetCommandQueueInfo(queue, CL_QUEUE_DEVICE, sizeof(cl_device_id), &device_id, NULL);
|
||||
if(err)
|
||||
return err;
|
||||
|
||||
size_t gSize;
|
||||
err = clGetKernelWorkGroupInfo(plan->twist_kernel, device_id, CL_KERNEL_WORK_GROUP_SIZE, sizeof(size_t), &gSize, NULL);
|
||||
if(err)
|
||||
return err;
|
||||
|
||||
gSize = min(128, gSize);
|
||||
size_t numGlobalThreads[1] = { max(numCols / gSize, 1)*gSize };
|
||||
size_t numLocalThreads[1] = { gSize };
|
||||
|
||||
err |= clSetKernelArg(plan->twist_kernel, 0, sizeof(cl_mem), &array_real);
|
||||
err |= clSetKernelArg(plan->twist_kernel, 1, sizeof(cl_mem), &array_imag);
|
||||
err |= clSetKernelArg(plan->twist_kernel, 2, sizeof(unsigned int), &sRow);
|
||||
err |= clSetKernelArg(plan->twist_kernel, 3, sizeof(unsigned int), &nCols);
|
||||
err |= clSetKernelArg(plan->twist_kernel, 4, sizeof(unsigned int), &N);
|
||||
err |= clSetKernelArg(plan->twist_kernel, 5, sizeof(unsigned int), &rToProcess);
|
||||
err |= clSetKernelArg(plan->twist_kernel, 6, sizeof(int), &d);
|
||||
|
||||
err |= clEnqueueNDRangeKernel(queue, plan->twist_kernel, 1, NULL, numGlobalThreads, numLocalThreads, 0, NULL, NULL);
|
||||
|
||||
return err;
|
||||
}
|
||||
|
||||
163
src/algorithms/libs/fft_internal.h
Normal file
163
src/algorithms/libs/fft_internal.h
Normal file
@@ -0,0 +1,163 @@
|
||||
|
||||
//
|
||||
// File: fft_internal.h
|
||||
//
|
||||
// Version: <1.0>
|
||||
//
|
||||
// Disclaimer: IMPORTANT: This Apple software is supplied to you by Apple Inc. ("Apple")
|
||||
// in consideration of your agreement to the following terms, and your use,
|
||||
// installation, modification or redistribution of this Apple software
|
||||
// constitutes acceptance of these terms. If you do not agree with these
|
||||
// terms, please do not use, install, modify or redistribute this Apple
|
||||
// software.
|
||||
//
|
||||
// In consideration of your agreement to abide by the following terms, and
|
||||
// subject to these terms, Apple grants you a personal, non - exclusive
|
||||
// license, under Apple's copyrights in this original Apple software ( the
|
||||
// "Apple Software" ), to use, reproduce, modify and redistribute the Apple
|
||||
// Software, with or without modifications, in source and / or binary forms;
|
||||
// provided that if you redistribute the Apple Software in its entirety and
|
||||
// without modifications, you must retain this notice and the following text
|
||||
// and disclaimers in all such redistributions of the Apple Software. Neither
|
||||
// the name, trademarks, service marks or logos of Apple Inc. may be used to
|
||||
// endorse or promote products derived from the Apple Software without specific
|
||||
// prior written permission from Apple. Except as expressly stated in this
|
||||
// notice, no other rights or licenses, express or implied, are granted by
|
||||
// Apple herein, including but not limited to any patent rights that may be
|
||||
// infringed by your derivative works or by other works in which the Apple
|
||||
// Software may be incorporated.
|
||||
//
|
||||
// The Apple Software is provided by Apple on an "AS IS" basis. APPLE MAKES NO
|
||||
// WARRANTIES, EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED
|
||||
// WARRANTIES OF NON - INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A
|
||||
// PARTICULAR PURPOSE, REGARDING THE APPLE SOFTWARE OR ITS USE AND OPERATION
|
||||
// ALONE OR IN COMBINATION WITH YOUR PRODUCTS.
|
||||
//
|
||||
// IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL OR
|
||||
// CONSEQUENTIAL DAMAGES ( INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION ) ARISING IN ANY WAY OUT OF THE USE, REPRODUCTION, MODIFICATION
|
||||
// AND / OR DISTRIBUTION OF THE APPLE SOFTWARE, HOWEVER CAUSED AND WHETHER
|
||||
// UNDER THEORY OF CONTRACT, TORT ( INCLUDING NEGLIGENCE ), STRICT LIABILITY OR
|
||||
// OTHERWISE, EVEN IF APPLE HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
//
|
||||
// Copyright ( C ) 2008 Apple Inc. All Rights Reserved.
|
||||
//
|
||||
////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
|
||||
#ifndef __CLFFT_INTERNAL_H
|
||||
#define __CLFFT_INTERNAL_H
|
||||
|
||||
#include "clFFT.h"
|
||||
#include <iostream>
|
||||
#include <string>
|
||||
#include <sstream>
|
||||
|
||||
using namespace std;
|
||||
|
||||
typedef enum kernel_dir_t
|
||||
{
|
||||
cl_fft_kernel_x,
|
||||
cl_fft_kernel_y,
|
||||
cl_fft_kernel_z
|
||||
}cl_fft_kernel_dir;
|
||||
|
||||
typedef struct kernel_info_t
|
||||
{
|
||||
cl_kernel kernel;
|
||||
char *kernel_name;
|
||||
unsigned lmem_size;
|
||||
unsigned num_workgroups;
|
||||
unsigned num_xforms_per_workgroup;
|
||||
unsigned num_workitems_per_workgroup;
|
||||
cl_fft_kernel_dir dir;
|
||||
int in_place_possible;
|
||||
kernel_info_t *next;
|
||||
}cl_fft_kernel_info;
|
||||
|
||||
typedef struct
|
||||
{
|
||||
// context in which fft resources are created and kernels are executed
|
||||
cl_context context;
|
||||
|
||||
// size of signal
|
||||
clFFT_Dim3 n;
|
||||
|
||||
// dimension of transform ... must be either 1D, 2D or 3D
|
||||
clFFT_Dimension dim;
|
||||
|
||||
// data format ... must be either interleaved or plannar
|
||||
clFFT_DataFormat format;
|
||||
|
||||
// string containing kernel source. Generated at runtime based on
|
||||
// n, dim, format and other parameters
|
||||
string *kernel_string;
|
||||
|
||||
// CL program containing source and kernel this particular
|
||||
// n, dim, data format
|
||||
cl_program program;
|
||||
|
||||
// linked list of kernels which needs to be executed for this fft
|
||||
cl_fft_kernel_info *kernel_info;
|
||||
|
||||
// number of kernels
|
||||
int num_kernels;
|
||||
|
||||
// twist kernel for virtualizing fft of very large sizes that do not
|
||||
// fit in GPU global memory
|
||||
cl_kernel twist_kernel;
|
||||
|
||||
// flag indicating if temporary intermediate buffer is needed or not.
|
||||
// this depends on fft kernels being executed and if transform is
|
||||
// in-place or out-of-place. e.g. Local memory fft (say 1D 1024 ...
|
||||
// one that does not require global transpose do not need temporary buffer)
|
||||
// 2D 1024x1024 out-of-place fft however do require intermediate buffer.
|
||||
// If temp buffer is needed, its allocation is lazy i.e. its not allocated
|
||||
// until its needed
|
||||
cl_int temp_buffer_needed;
|
||||
|
||||
// Batch size is runtime parameter and size of temporary buffer (if needed)
|
||||
// depends on batch size. Allocation of temporary buffer is lazy i.e. its
|
||||
// only created when needed. Once its created at first call of clFFT_Executexxx
|
||||
// it is not allocated next time if next time clFFT_Executexxx is called with
|
||||
// batch size different than the first call. last_batch_size caches the last
|
||||
// batch size with which this plan is used so that we dont keep allocating/deallocating
|
||||
// temp buffer if same batch size is used again and again.
|
||||
unsigned last_batch_size;
|
||||
|
||||
// temporary buffer for interleaved plan
|
||||
cl_mem tempmemobj;
|
||||
|
||||
// temporary buffer for planner plan. Only one of tempmemobj or
|
||||
// (tempmemobj_real, tempmemobj_imag) pair is valid (allocated) depending
|
||||
// data format of plan (plannar or interleaved)
|
||||
cl_mem tempmemobj_real, tempmemobj_imag;
|
||||
|
||||
// Maximum size of signal for which local memory transposed based
|
||||
// fft is sufficient i.e. no global mem transpose (communication)
|
||||
// is needed
|
||||
unsigned max_localmem_fft_size;
|
||||
|
||||
// Maximum work items per work group allowed. This, along with max_radix below controls
|
||||
// maximum local memory being used by fft kernels of this plan. Set to 256 by default
|
||||
unsigned max_work_item_per_workgroup;
|
||||
|
||||
// Maximum base radix for local memory fft ... this controls the maximum register
|
||||
// space used by work items. Currently defaults to 16
|
||||
unsigned max_radix;
|
||||
|
||||
// Device depended parameter that tells how many work-items need to be read consecutive
|
||||
// values to make sure global memory access by work-items of a work-group result in
|
||||
// coalesced memory access to utilize full bandwidth e.g. on NVidia tesla, this is 16
|
||||
unsigned min_mem_coalesce_width;
|
||||
|
||||
// Number of local memory banks. This is used to geneate kernel with local memory
|
||||
// transposes with appropriate padding to avoid bank conflicts to local memory
|
||||
// e.g. on NVidia it is 16.
|
||||
unsigned num_local_mem_banks;
|
||||
}cl_fft_plan;
|
||||
|
||||
void FFT1D(cl_fft_plan *plan, cl_fft_kernel_dir dir);
|
||||
|
||||
#endif
|
||||
1257
src/algorithms/libs/fft_kernelstring.cc
Normal file
1257
src/algorithms/libs/fft_kernelstring.cc
Normal file
File diff suppressed because it is too large
Load Diff
402
src/algorithms/libs/fft_setup.cc
Normal file
402
src/algorithms/libs/fft_setup.cc
Normal file
@@ -0,0 +1,402 @@
|
||||
|
||||
//
|
||||
// File: fft_setup.cpp
|
||||
//
|
||||
// Version: <1.0>
|
||||
//
|
||||
// Disclaimer: IMPORTANT: This Apple software is supplied to you by Apple Inc. ("Apple")
|
||||
// in consideration of your agreement to the following terms, and your use,
|
||||
// installation, modification or redistribution of this Apple software
|
||||
// constitutes acceptance of these terms. If you do not agree with these
|
||||
// terms, please do not use, install, modify or redistribute this Apple
|
||||
// software.
|
||||
//
|
||||
// In consideration of your agreement to abide by the following terms, and
|
||||
// subject to these terms, Apple grants you a personal, non - exclusive
|
||||
// license, under Apple's copyrights in this original Apple software ( the
|
||||
// "Apple Software" ), to use, reproduce, modify and redistribute the Apple
|
||||
// Software, with or without modifications, in source and / or binary forms;
|
||||
// provided that if you redistribute the Apple Software in its entirety and
|
||||
// without modifications, you must retain this notice and the following text
|
||||
// and disclaimers in all such redistributions of the Apple Software. Neither
|
||||
// the name, trademarks, service marks or logos of Apple Inc. may be used to
|
||||
// endorse or promote products derived from the Apple Software without specific
|
||||
// prior written permission from Apple. Except as expressly stated in this
|
||||
// notice, no other rights or licenses, express or implied, are granted by
|
||||
// Apple herein, including but not limited to any patent rights that may be
|
||||
// infringed by your derivative works or by other works in which the Apple
|
||||
// Software may be incorporated.
|
||||
//
|
||||
// The Apple Software is provided by Apple on an "AS IS" basis. APPLE MAKES NO
|
||||
// WARRANTIES, EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED
|
||||
// WARRANTIES OF NON - INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A
|
||||
// PARTICULAR PURPOSE, REGARDING THE APPLE SOFTWARE OR ITS USE AND OPERATION
|
||||
// ALONE OR IN COMBINATION WITH YOUR PRODUCTS.
|
||||
//
|
||||
// IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL OR
|
||||
// CONSEQUENTIAL DAMAGES ( INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION ) ARISING IN ANY WAY OUT OF THE USE, REPRODUCTION, MODIFICATION
|
||||
// AND / OR DISTRIBUTION OF THE APPLE SOFTWARE, HOWEVER CAUSED AND WHETHER
|
||||
// UNDER THEORY OF CONTRACT, TORT ( INCLUDING NEGLIGENCE ), STRICT LIABILITY OR
|
||||
// OTHERWISE, EVEN IF APPLE HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
//
|
||||
// Copyright ( C ) 2008 Apple Inc. All Rights Reserved.
|
||||
//
|
||||
////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
|
||||
#include "fft_internal.h"
|
||||
#include "fft_base_kernels.h"
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
#include <iostream>
|
||||
#include <string>
|
||||
#include <sstream>
|
||||
#include <limits>
|
||||
|
||||
using namespace std;
|
||||
|
||||
extern void getKernelWorkDimensions(cl_fft_plan *plan, cl_fft_kernel_info *kernelInfo, cl_int *batchSize, size_t *gWorkItems, size_t *lWorkItems);
|
||||
|
||||
static void
|
||||
getBlockConfigAndKernelString(cl_fft_plan *plan)
|
||||
{
|
||||
plan->temp_buffer_needed = 0;
|
||||
*plan->kernel_string += baseKernels;
|
||||
|
||||
if(plan->format == clFFT_SplitComplexFormat)
|
||||
*plan->kernel_string += twistKernelPlannar;
|
||||
else
|
||||
*plan->kernel_string += twistKernelInterleaved;
|
||||
|
||||
switch(plan->dim)
|
||||
{
|
||||
case clFFT_1D:
|
||||
FFT1D(plan, cl_fft_kernel_x);
|
||||
break;
|
||||
|
||||
case clFFT_2D:
|
||||
FFT1D(plan, cl_fft_kernel_x);
|
||||
FFT1D(plan, cl_fft_kernel_y);
|
||||
break;
|
||||
|
||||
case clFFT_3D:
|
||||
FFT1D(plan, cl_fft_kernel_x);
|
||||
FFT1D(plan, cl_fft_kernel_y);
|
||||
FFT1D(plan, cl_fft_kernel_z);
|
||||
break;
|
||||
|
||||
default:
|
||||
return;
|
||||
}
|
||||
|
||||
plan->temp_buffer_needed = 0;
|
||||
cl_fft_kernel_info *kInfo = plan->kernel_info;
|
||||
while(kInfo)
|
||||
{
|
||||
plan->temp_buffer_needed |= !kInfo->in_place_possible;
|
||||
kInfo = kInfo->next;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
static void
|
||||
deleteKernelInfo(cl_fft_kernel_info *kInfo)
|
||||
{
|
||||
if(kInfo)
|
||||
{
|
||||
if(kInfo->kernel_name)
|
||||
free(kInfo->kernel_name);
|
||||
if(kInfo->kernel)
|
||||
clReleaseKernel(kInfo->kernel);
|
||||
free(kInfo);
|
||||
}
|
||||
}
|
||||
|
||||
static void
|
||||
destroy_plan(cl_fft_plan *Plan)
|
||||
{
|
||||
cl_fft_kernel_info *kernel_info = Plan->kernel_info;
|
||||
|
||||
while(kernel_info)
|
||||
{
|
||||
cl_fft_kernel_info *tmp = kernel_info->next;
|
||||
deleteKernelInfo(kernel_info);
|
||||
kernel_info = tmp;
|
||||
}
|
||||
|
||||
Plan->kernel_info = NULL;
|
||||
|
||||
if(Plan->kernel_string)
|
||||
{
|
||||
delete Plan->kernel_string;
|
||||
Plan->kernel_string = NULL;
|
||||
}
|
||||
if(Plan->twist_kernel)
|
||||
{
|
||||
clReleaseKernel(Plan->twist_kernel);
|
||||
Plan->twist_kernel = NULL;
|
||||
}
|
||||
if(Plan->program)
|
||||
{
|
||||
clReleaseProgram(Plan->program);
|
||||
Plan->program = NULL;
|
||||
}
|
||||
if(Plan->tempmemobj)
|
||||
{
|
||||
clReleaseMemObject(Plan->tempmemobj);
|
||||
Plan->tempmemobj = NULL;
|
||||
}
|
||||
if(Plan->tempmemobj_real)
|
||||
{
|
||||
clReleaseMemObject(Plan->tempmemobj_real);
|
||||
Plan->tempmemobj_real = NULL;
|
||||
}
|
||||
if(Plan->tempmemobj_imag)
|
||||
{
|
||||
clReleaseMemObject(Plan->tempmemobj_imag);
|
||||
Plan->tempmemobj_imag = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
static int
|
||||
createKernelList(cl_fft_plan *plan)
|
||||
{
|
||||
cl_program program = plan->program;
|
||||
cl_fft_kernel_info *kernel_info = plan->kernel_info;
|
||||
|
||||
cl_int err;
|
||||
while(kernel_info)
|
||||
{
|
||||
kernel_info->kernel = clCreateKernel(program, kernel_info->kernel_name, &err);
|
||||
if(!kernel_info->kernel || err != CL_SUCCESS)
|
||||
return err;
|
||||
kernel_info = kernel_info->next;
|
||||
}
|
||||
|
||||
if(plan->format == clFFT_SplitComplexFormat)
|
||||
plan->twist_kernel = clCreateKernel(program, "clFFT_1DTwistSplit", &err);
|
||||
else
|
||||
plan->twist_kernel = clCreateKernel(program, "clFFT_1DTwistInterleaved", &err);
|
||||
|
||||
if(!plan->twist_kernel || err)
|
||||
return err;
|
||||
|
||||
return CL_SUCCESS;
|
||||
}
|
||||
|
||||
int getMaxKernelWorkGroupSize(cl_fft_plan *plan, unsigned int *max_wg_size, unsigned int num_devices, cl_device_id *devices)
|
||||
{
|
||||
int reg_needed = 0;
|
||||
*max_wg_size = std::numeric_limits<int>::max();
|
||||
int err;
|
||||
unsigned wg_size;
|
||||
|
||||
unsigned int i;
|
||||
for(i = 0; i < num_devices; i++)
|
||||
{
|
||||
cl_fft_kernel_info *kInfo = plan->kernel_info;
|
||||
while(kInfo)
|
||||
{
|
||||
err = clGetKernelWorkGroupInfo(kInfo->kernel, devices[i], CL_KERNEL_WORK_GROUP_SIZE, sizeof(size_t), &wg_size, NULL);
|
||||
if(err != CL_SUCCESS)
|
||||
return -1;
|
||||
|
||||
if(wg_size < kInfo->num_workitems_per_workgroup)
|
||||
reg_needed |= 1;
|
||||
|
||||
if(*max_wg_size > wg_size)
|
||||
*max_wg_size = wg_size;
|
||||
|
||||
kInfo = kInfo->next;
|
||||
}
|
||||
}
|
||||
|
||||
return reg_needed;
|
||||
}
|
||||
|
||||
#define ERR_MACRO(err) { \
|
||||
if( err != CL_SUCCESS) \
|
||||
{ \
|
||||
if(error_code) \
|
||||
*error_code = err; \
|
||||
clFFT_DestroyPlan((clFFT_Plan) plan); \
|
||||
return (clFFT_Plan) NULL; \
|
||||
} \
|
||||
}
|
||||
|
||||
clFFT_Plan
|
||||
clFFT_CreatePlan(cl_context context, clFFT_Dim3 n, clFFT_Dimension dim, clFFT_DataFormat dataFormat, cl_int *error_code )
|
||||
{
|
||||
int i;
|
||||
cl_int err;
|
||||
int isPow2 = 1;
|
||||
cl_fft_plan *plan = NULL;
|
||||
ostringstream kString;
|
||||
int num_devices;
|
||||
int gpu_found = 0;
|
||||
cl_device_id devices[16];
|
||||
size_t ret_size;
|
||||
cl_device_type device_type;
|
||||
|
||||
if(!context)
|
||||
ERR_MACRO(CL_INVALID_VALUE);
|
||||
|
||||
isPow2 |= n.x && !( (n.x - 1) & n.x );
|
||||
isPow2 |= n.y && !( (n.y - 1) & n.y );
|
||||
isPow2 |= n.z && !( (n.z - 1) & n.z );
|
||||
|
||||
if(!isPow2)
|
||||
ERR_MACRO(CL_INVALID_VALUE);
|
||||
|
||||
if( (dim == clFFT_1D && (n.y != 1 || n.z != 1)) || (dim == clFFT_2D && n.z != 1) )
|
||||
ERR_MACRO(CL_INVALID_VALUE);
|
||||
|
||||
plan = (cl_fft_plan *) malloc(sizeof(cl_fft_plan));
|
||||
if(!plan)
|
||||
ERR_MACRO(CL_OUT_OF_RESOURCES);
|
||||
|
||||
plan->context = context;
|
||||
clRetainContext(context);
|
||||
plan->n = n;
|
||||
plan->dim = dim;
|
||||
plan->format = dataFormat;
|
||||
plan->kernel_info = 0;
|
||||
plan->num_kernels = 0;
|
||||
plan->twist_kernel = 0;
|
||||
plan->program = 0;
|
||||
plan->temp_buffer_needed = 0;
|
||||
plan->last_batch_size = 0;
|
||||
plan->tempmemobj = 0;
|
||||
plan->tempmemobj_real = 0;
|
||||
plan->tempmemobj_imag = 0;
|
||||
plan->max_localmem_fft_size = 2048;
|
||||
plan->max_work_item_per_workgroup = 256;
|
||||
plan->max_radix = 16;
|
||||
plan->min_mem_coalesce_width = 16;
|
||||
plan->num_local_mem_banks = 16;
|
||||
|
||||
patch_kernel_source:
|
||||
|
||||
plan->kernel_string = new string("");
|
||||
if(!plan->kernel_string)
|
||||
ERR_MACRO(CL_OUT_OF_RESOURCES);
|
||||
|
||||
getBlockConfigAndKernelString(plan);
|
||||
|
||||
const char *source_str = plan->kernel_string->c_str();
|
||||
plan->program = clCreateProgramWithSource(context, 1, (const char**) &source_str, NULL, &err);
|
||||
ERR_MACRO(err);
|
||||
|
||||
err = clGetContextInfo(context, CL_CONTEXT_DEVICES, sizeof(devices), devices, &ret_size);
|
||||
ERR_MACRO(err);
|
||||
|
||||
num_devices = (int)(ret_size / sizeof(cl_device_id));
|
||||
|
||||
for(i = 0; i < num_devices; i++)
|
||||
{
|
||||
err = clGetDeviceInfo(devices[i], CL_DEVICE_TYPE, sizeof(device_type), &device_type, NULL);
|
||||
ERR_MACRO(err);
|
||||
|
||||
if(device_type == CL_DEVICE_TYPE_GPU)
|
||||
{
|
||||
gpu_found = 1;
|
||||
err = clBuildProgram(plan->program, 1, &devices[i], "-cl-mad-enable", NULL, NULL);
|
||||
if (err != CL_SUCCESS)
|
||||
{
|
||||
char *build_log;
|
||||
char devicename[200];
|
||||
size_t log_size;
|
||||
|
||||
err = clGetProgramBuildInfo(plan->program, devices[i], CL_PROGRAM_BUILD_LOG, 0, NULL, &log_size);
|
||||
ERR_MACRO(err);
|
||||
|
||||
build_log = (char *) malloc(log_size + 1);
|
||||
|
||||
err = clGetProgramBuildInfo(plan->program, devices[i], CL_PROGRAM_BUILD_LOG, log_size, build_log, NULL);
|
||||
ERR_MACRO(err);
|
||||
|
||||
err = clGetDeviceInfo(devices[i], CL_DEVICE_NAME, sizeof(devicename), devicename, NULL);
|
||||
ERR_MACRO(err);
|
||||
|
||||
fprintf(stdout, "FFT program build log on device %s\n", devicename);
|
||||
fprintf(stdout, "%s\n", build_log);
|
||||
free(build_log);
|
||||
|
||||
ERR_MACRO(err);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if(!gpu_found)
|
||||
ERR_MACRO(CL_INVALID_CONTEXT);
|
||||
|
||||
err = createKernelList(plan);
|
||||
ERR_MACRO(err);
|
||||
|
||||
// we created program and kernels based on "some max work group size (default 256)" ... this work group size
|
||||
// may be larger than what kernel may execute with ... if thats the case we need to regenerate the kernel source
|
||||
// setting this as limit i.e max group size and rebuild.
|
||||
unsigned int max_kernel_wg_size;
|
||||
int patching_req = getMaxKernelWorkGroupSize(plan, &max_kernel_wg_size, num_devices, devices);
|
||||
if(patching_req == -1)
|
||||
{
|
||||
ERR_MACRO(err);
|
||||
}
|
||||
|
||||
if(patching_req)
|
||||
{
|
||||
destroy_plan(plan);
|
||||
plan->max_work_item_per_workgroup = max_kernel_wg_size;
|
||||
goto patch_kernel_source;
|
||||
}
|
||||
|
||||
cl_fft_kernel_info *kInfo = plan->kernel_info;
|
||||
while(kInfo)
|
||||
{
|
||||
plan->num_kernels++;
|
||||
kInfo = kInfo->next;
|
||||
}
|
||||
|
||||
if(error_code)
|
||||
*error_code = CL_SUCCESS;
|
||||
|
||||
return (clFFT_Plan) plan;
|
||||
}
|
||||
|
||||
void
|
||||
clFFT_DestroyPlan(clFFT_Plan plan)
|
||||
{
|
||||
cl_fft_plan *Plan = (cl_fft_plan *) plan;
|
||||
if(Plan)
|
||||
{
|
||||
destroy_plan(Plan);
|
||||
clReleaseContext(Plan->context);
|
||||
free(Plan);
|
||||
}
|
||||
}
|
||||
|
||||
void clFFT_DumpPlan( clFFT_Plan Plan, FILE *file)
|
||||
{
|
||||
size_t gDim, lDim;
|
||||
FILE *out;
|
||||
if(!file)
|
||||
out = stdout;
|
||||
else
|
||||
out = file;
|
||||
|
||||
cl_fft_plan *plan = (cl_fft_plan *) Plan;
|
||||
cl_fft_kernel_info *kInfo = plan->kernel_info;
|
||||
|
||||
while(kInfo)
|
||||
{
|
||||
cl_int s = 1;
|
||||
getKernelWorkDimensions(plan, kInfo, &s, &gDim, &lDim);
|
||||
fprintf(out, "Run kernel %s with global dim = {%zd*BatchSize}, local dim={%zd}\n", kInfo->kernel_name, gDim, lDim);
|
||||
kInfo = kInfo->next;
|
||||
}
|
||||
fprintf(out, "%s\n", plan->kernel_string->c_str());
|
||||
}
|
||||
@@ -158,7 +158,7 @@ galileo_e1_code_gen_complex_sampled(std::complex<float>* _dest, char _Signal[3],
|
||||
|
||||
std::string _galileo_signal = _Signal;
|
||||
unsigned int _samplesPerCode;
|
||||
const unsigned int _codeFreqBasis = Galileo_E1_CODE_CHIP_RATE_HZ; //Hz
|
||||
const int _codeFreqBasis = Galileo_E1_CODE_CHIP_RATE_HZ; //Hz
|
||||
unsigned int _codeLength = Galileo_E1_B_CODE_LENGTH_CHIPS;
|
||||
int primary_code_E1_chips[(int)Galileo_E1_B_CODE_LENGTH_CHIPS];
|
||||
_samplesPerCode = round(_fs / (_codeFreqBasis / _codeLength));
|
||||
|
||||
Reference in New Issue
Block a user