Fix the reduce GPU test in 2D.