Switch to GPU aware MPI transfer and manual reduction on GPU