Function gradient

src/backend/cuda/kernel/gradient.hpp:25–48

template<typename T>
void gradient(Param<T> grad0, Param<T> grad1, CParam<T> in) {
    constexpr unsigned TX = 32;
    constexpr unsigned TY = 8;

    auto gradient =
        common::getKernel("arrayfire::cuda::gradient", {{gradient_cuh_src}},
                          TemplateArgs(TemplateTypename<T>()),
                          {{DefineValue(TX), DefineValue(TY)}});

    dim3 threads(TX, TY, 1);

    int blocksPerMatX = divup(in.dims[0], TX);
    int blocksPerMatY = divup(in.dims[1], TY);
    dim3 blocks(blocksPerMatX * in.dims[2], blocksPerMatY * in.dims[3], 1);

    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());

    gradient(qArgs, grad0, grad1, in, blocksPerMatX, blocksPerMatY);
    POST_LAUNCH_CHECK();
}

}  // namespace kernel
}  // namespace cuda