![]() |
Stan Math Library
5.1.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.