fft.cl 706 Bytes
Newer Older
Peter Eastman's avatar
Peter Eastman committed
1
2
3
4
5
6
7
8
9
10
float2 multiplyComplex(float2 c1, float2 c2) {
    return (float2) (c1.x*c2.x-c1.y*c2.y, c1.x*c2.y+c1.y*c2.x);
}

/**
 * Perform a 1D FFT on each row along one axis.
 */

__kernel void execFFT(__global float2* matrix, float sign, __local float2* w, __local float2* data0, __local float2* data1) {
    const int i = get_local_id(0);
Peter Eastman's avatar
Peter Eastman committed
11
    w[i] = (float2) (cos(-sign*i*2*M_PI/XSIZE), sin(-sign*i*2*M_PI/XSIZE));
12
    for (int index = get_group_id(0); index < YSIZE*ZSIZE; index += get_num_groups(0)) {
Peter Eastman's avatar
Peter Eastman committed
13
14
15
16
17
18
19
        int z = index/YSIZE;
        int y = index-z*YSIZE;
        int element = i*XMULT+y*YMULT+z*ZMULT;
        data0[i] = matrix[element];
        barrier(CLK_LOCAL_MEM_FENCE);
        COMPUTE_FFT
    }
Peter Eastman's avatar
Peter Eastman committed
20
}