Mercurial > hg > Game > Cerium
view example/fft/cuda/norm.cu @ 2069:26aa08c9a1de draft default tip
cuda example fix
author | Shinji KONO <kono@ie.u-ryukyu.ac.jp> |
---|---|
date | Sun, 12 Feb 2017 10:04:55 +0900 |
parents | 433043c56a0c |
children |
line wrap: on
line source
extern "C" { __global__ void norm(long* param, float* x) { unsigned long gid = blockIdx.x*blockDim.x+threadIdx.x; // (unsigned long)s->get_param(0); unsigned long nid = blockIdx.y*blockDim.y+threadIdx.y; //(unsigned long)s->get_param(1); long n = param[0]; x[(nid*n+gid)*2] = x[(nid*n+gid)*2] / (float)n; x[(nid*n+gid)*2+1] = x[(nid*n+gid)*2+1] / (float)n; } }