Stan Math Library
5.0.0
Automatic Differentiation
|
__kernel void stan::math::opencl_kernels::cumulative_sum1 | ( | __global SCAL * | out_wgs, |
__global SCAL * | out_threads, | ||
__global SCAL * | in, | ||
int | size | ||
) |
First kernel of the cumulative sum implementation.
Each thread sums the assigned elements and threads within same work group add their results together.
[out] | out_wgs | results from each work group |
[out] | out_threads | results for each thread |
[in] | in | input data |
size | size number of elements in the input |
Definition at line 27 of file cumulative_sum.hpp.