コード例 #1
0
ファイル: DataChannelMPI.cpp プロジェクト: xiongyw/pytorch
void DataChannelMPI::receive(Scalar& data, rank_type src_rank) {
  MPI_Recv(data.data(), data.elementSize(), MPI_UINT8_T,
           src_rank, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
}
コード例 #2
0
ファイル: DataChannelMPI.cpp プロジェクト: xiongyw/pytorch
void DataChannelMPI::send(Scalar& data, rank_type dst_rank) {
  MPI_Send(data.data(), data.elementSize(), MPI_UINT8_T,
           dst_rank, 0, MPI_COMM_WORLD);
}