Avoid CDF overreads in gather_top_partition_prob()

This makes experimenting with MSAC easier due to not having a weird special requirement in this one function. Also technically an optimization.