with MPI. This will work for one node with several devices and one MPI
process per device.

* Now messages_info has the all_nodes optional argument.