Added support for cuda aware MPI, does not work yet