MPI collective communication with subarrays working now