Function gradient

src/backend/opencl/kernel/gradient.hpp:29–58

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

    std::array<TemplateArg, 1> targs = {
        TemplateTypename<T>(),
    };
    std::array<std::string, 6> options = {
        DefineKeyValue(T, dtype_traits<T>::getName()),
        DefineValue(TX),
        DefineValue(TY),
        DefineKeyValue(ZERO, scalar_to_option(scalar<T>(0))),
        DefineKeyValue(CPLX, static_cast<int>(iscplx<T>())),
        getTypeBuildDefinition<T>()};

    auto gradOp =
        common::getKernel("gradient", {{gradient_cl_src}}, targs, options);

    cl::NDRange local(TX, TY, 1);

    int blocksPerMatX = divup(in.info.dims[0], TX);
    int blocksPerMatY = divup(in.info.dims[1], TY);
    cl::NDRange global(local[0] * blocksPerMatX * in.info.dims[2],
                       local[1] * blocksPerMatY * in.info.dims[3], 1);

    gradOp(cl::EnqueueArgs(getQueue(), global, local), *grad0.data, grad0.info,
           *grad1.data, grad1.info, *in.data, in.info, blocksPerMatX,
           blocksPerMatY);
    CL_DEBUG_FINISH(getQueue());
}
}  // namespace kernel
}  // namespace opencl
}  // namespace arrayfire