view example/fft/ppe/norm.cc @ 1658:ed2fa50168ef draft

fix fft
author Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
date Tue, 09 Jul 2013 19:15:56 +0900
parents 6c0b6947c231
children 724fd3c1ac92
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_inputAddr(0);

    cl_float2* out_x = (cl_float2*)s->get_outputAddr(0);

    unsigned long gid = (unsigned long)s->get_param(0);
    unsigned long nid = (unsigned long)s->get_param(1);

    long n = (long)s->get_param(3);

    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;
}