Commit 2f44059b authored by Tomáš Oberhuber's avatar Tomáš Oberhuber
Browse files

MpiCommunicator cheks MPI_CUDA awarness only running really in more MPI processes.

parent e053b414

src/TNL/Communicators/MpiCommunicator.h

+11 −6
Original line number Diff line number Diff line
@@ -105,6 +105,10 @@ class MpiCommunicator 
            redirect = parameters.getParameter< bool >( "redirect-mpi-output" );

            setupRedirection();

#ifdef HAVE_CUDA

            int size;

            MPI_Comm_size( MPI_COMM_WORLD, &size );

            if( size > 1 )

            {

   #if defined( MPIX_CUDA_AWARE_SUPPORT ) && MPIX_CUDA_AWARE_SUPPORT

               std::cout << "CUDA-aware MPI detected on this system ... " << std::endl;

   #elif defined( MPIX_CUDA_AWARE_SUPPORT ) && !MPIX_CUDA_AWARE_SUPPORT
@@ -113,6 +117,7 @@ class MpiCommunicator 
   #else

               std::cerr << "WARNING: TNL cannot detect if you have CUDA-aware MPI. Some problems may occur." << std::endl;

   #endif

            }

#endif // HAVE_CUDA

            bool gdbDebug = parameters.getParameter< bool >( "mpi-gdb-debug" );

            int processToAttach = parameters.getParameter< int >( "mpi-process-to-attach" );