view example/fft/ppe/norm.cc @ 1817:b0376e1c51e9 draft

use get_input & get_output
author Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
date Fri, 13 Dec 2013 05:18:58 +0900
parents 614c60736bfd
children 144e573b030b
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(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;
}