add reduce_only option to prefix_sum_ops