Home
last modified time | relevance | path

Searched refs:MPI_CHECK (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupMPI.cpp17 #define MPI_CHECK(cmd) \ macro
147 MPI_CHECK(MPI_Test(&request_, &flag, &status_)); in isCompleted()
187 MPI_CHECK(MPI_Wait(&request_, &status_)); in wait()
215 MPI_CHECK(MPI_Error_string(status_.MPI_ERROR, buf.data(), &len)); in populateException()
228 MPI_CHECK(MPI_Finalize()); in mpiExit()
235 MPI_CHECK(MPI_Initialized(&mpi_was_initialized)); in initMPIOnce()
237 MPI_CHECK(MPI_Init_thread( in initMPIOnce()
272 MPI_CHECK(MPI_Comm_group(MPI_COMM_WORLD, &worldGroup)); in createProcessGroupMPI()
273 MPI_CHECK( in createProcessGroupMPI()
287 MPI_CHECK(groupComm_updated); in createProcessGroupMPI()
[all …]