Function unwrap

src/backend/cuda/kernel/unwrap.hpp:24–56

template<typename T>
void unwrap(Param<T> out, CParam<T> in, const int wx, const int wy,
            const int sx, const int sy, const int px, const int py,
            const int dx, const int dy, const int nx, const bool is_column) {
    auto unwrap = common::getKernel(
        "arrayfire::cuda::unwrap", {{unwrap_cuh_src}},
        TemplateArgs(TemplateTypename<T>(), TemplateArg(is_column)));

    dim3 threads, blocks;
    int reps;

    if (is_column) {
        int TX = std::min(THREADS_PER_BLOCK, nextpow2(out.dims[0]));

        threads = dim3(TX, THREADS_PER_BLOCK / TX);
        blocks = dim3(divup(out.dims[1], threads.y), out.dims[2] * out.dims[3]);
        reps   = divup((wx * wy),
                       threads.x);  // is > 1 only when TX == 256 && wx * wy > 256
    } else {
        threads = dim3(THREADS_X, THREADS_Y);
        blocks = dim3(divup(out.dims[0], threads.x), out.dims[2] * out.dims[3]);

        reps = divup((wx * wy), threads.y);
    }

    const int maxBlocksY = getDeviceProp(getActiveDeviceId()).maxGridSize[1];
    blocks.z             = divup(blocks.y, maxBlocksY);
    blocks.y             = divup(blocks.y, blocks.z);

    EnqueueArgs qArgs(blocks, threads, getActiveStream());

    unwrap(qArgs, out, in, wx, wy, sx, sy, px, py, dx, dy, nx, reps);
    POST_LAUNCH_CHECK();
}

}  // namespace kernel
}  // namespace cuda