Function EncodeStreamIdToInt64

oneflow/core/graph/stream_id.cpp:48–54

}  // namespace

int64_t EncodeStreamIdToInt64(const StreamId& stream_id) {
  int64_t id = static_cast<int64_t>(stream_id.stream_index());
  id |= static_cast<int64_t>(stream_id.device_index()) << kDeviceIndexShift;
  id |= static_cast<int64_t>(stream_id.device_type()) << kDeviceTypeShift;
  id |= static_cast<int64_t>(stream_id.rank()) << kRankShift;
  return id;
}

StreamId DecodeStreamIdFromInt64(int64_t stream_id_val) {
  int64_t rank = (stream_id_val & kRankInt64Mask) >> kRankShift;