Fixed compile error when ENABLE_MPI_CUDA is ON