[opt warning] fix gpu reduce opt warning.

This allow to move more computation to the gpu.