/* ----------------------------------------------------------------------------- * * Kernel : Slice * Copyright : [2008..2011] Manuel M T Chakravarty, Gabriele Keller, Sean Lee, Trevor L. McDonell * License : BSD3 * * Maintainer : Trevor L. McDonell * Stability : experimental * * ---------------------------------------------------------------------------*/ extern "C" __global__ void slice ( ArrOut d_out, const ArrIn0 d_in0, const Slice slice, const CoSlice slix, const SliceDim sliceDim ) { const Ix shapeSize = size(slice); const Ix gridSize = __umul24(blockDim.x, gridDim.x); for (Ix ix = __umul24(blockDim.x, blockIdx.x) + threadIdx.x; ix < shapeSize; ix += gridSize) { Slice dst = fromIndex(slice, ix); SliceDim src = sliceIndex(dst, slix); set(d_out, ix, get0(d_in0, toIndex(sliceDim, src))); } }