Optimizations to multi-GPU calculations

It doesn't make a bit difference, but it is a little faster.