// @library_deps: nmath
// @depends_nmath: pgeom
// @all_depends_nmath_count: 7
// @all_depends_nmath: dpq, Rmath, nmath, chebyshev, log1p, expm1, pgeom

#pragma OPENCL EXTENSION cl_khr_fp64 : enable

__kernel void pgeom_kernel(
    __global const double* q,
    __global const double* prob,
    __global const int* lower_tail,
    __global const int* log_p,
    __global double* out,
    const int len
) {
    int i = get_global_id(0);
    if (i >= len) return;
    int lt = (lower_tail[i] != 0) ? 1 : 0;
    int lp = (log_p[i] != 0) ? 1 : 0;
    out[i] = pgeom(q[i], prob[i], lt, lp);
}
