Searched defs:SumReduceOp (Results 1 – 1 of 1) sorted by relevance
85 struct SumReduceOp { struct86 __device__ __forceinline__ acc_t combine(acc_t a, acc_t b) const { return a + b; } in combine()88 __device__ __forceinline__ acc_t warp_shfl_down(acc_t data, int offset) const { in warp_shfl_down()94 struct SumReduceOp<Float2<scalar_t, accscalar_t>> { struct95 using acc_t = Float2<scalar_t, accscalar_t>;97 __device__ __forceinline__ acc_t combine(acc_t a, acc_t b) const { return a + b; } in combine()99 __device__ __forceinline__ acc_t warp_shfl_down(acc_t data, int offset) const { in warp_shfl_down()