view example/fft/ppe/norm.cc @ 1835:144e573b030b draft

fix fft
author Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
date Fri, 20 Dec 2013 03:00:32 +0900
parents b0376e1c51e9
children
line wrap: on
line source

#include "norm.h"
#ifdef __APPLE__
#include <OpenCL/opencl.h>
#else
#include <CL/cl.h>
#endif
#include "Func.h"
SchedDefineTask1(norm,norm);

static int
norm(SchedTask* s, void* rbuf,void* wbuf)
{
    cl_float2* in_x = (cl_float2*)s->get_input(rbuf,0);

    cl_float2* out_x = (cl_float2*)s->get_output(wbuf,0);

    unsigned long gid = s->x; // (unsigned long)s->get_param(0);
    unsigned long nid = s->y; //(unsigned long)s->get_param(1);
    
    long n = (long)s->get_param(0);

    out_x[nid*n+gid].x = in_x[nid*n+gid].x / (float)n;
    out_x[nid*n+gid].y = in_x[nid*n+gid].y / (float)n;

    return 0;
}