Function triangle

src/backend/cuda/kernel/triangle.hpp:23–48

template<typename T>
void triangle(Param<T> r, CParam<T> in, bool is_upper, bool is_unit_diag) {
    constexpr unsigned TX    = 32;
    constexpr unsigned TY    = 8;
    constexpr unsigned TILEX = 128;
    constexpr unsigned TILEY = 32;

    auto triangle = common::getKernel(
        "arrayfire::cuda::triangle", {{triangle_cuh_src}},
        TemplateArgs(TemplateTypename<T>(), TemplateArg(is_upper),
                     TemplateArg(is_unit_diag)));

    dim3 threads(TX, TY, 1);

    int blocksPerMatX = divup(r.dims[0], TILEX);
    int blocksPerMatY = divup(r.dims[1], TILEY);
    dim3 blocks(blocksPerMatX * r.dims[2], blocksPerMatY * r.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());

    triangle(qArgs, r, in, blocksPerMatX, blocksPerMatY);
    POST_LAUNCH_CHECK();
}

}  // namespace kernel
}  // namespace cuda