Fix for CUDA partial halo exchange