C++ (Cpp) Scalar::dataの例

コード例 #1

ファイル: 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

ファイル: 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);
}