Fix partial reduction on CPU.