Improvements to multi-GPU performance