Automatic Differentiation
 
Loading...
Searching...
No Matches

◆ cumulative_sum3()

__kernel void stan::math::opencl_kernels::cumulative_sum3 ( __global SCAL *  out,
__global SCAL *  in_data,
__global SCAL *  in_threads,
__global SCAL *  in_wgs,
int  size 
)

Third kernel of the cumulative sum implementation.

Given sums of threads and cumulative sum of those calculates cumulative sum of given array. Must be run with the same number of threads and work groups as the first cumulative sum kernel.

Parameters
[out]outcumulatively summed input
[out]in_datainput data
[in]in_threadssummed results from each thread from the first kernel
[in]in_wgscumulatively summed results from each work group (calculated by previous two kernels)
sizesize number of elements in the input

Definition at line 135 of file cumulative_sum.hpp.