prefix_sum kernel in progress