Function EncodeTaskIdToInt64

oneflow/core/graph/task_id.cpp:52–59

}  // namespace

int64_t EncodeTaskIdToInt64(const TaskId& task_id) {
  int64_t id = static_cast<int64_t>(task_id.task_index());
  id |= static_cast<int64_t>(task_id.stream_id().stream_index()) << kStreamIndexShift;
  id |= static_cast<int64_t>(task_id.stream_id().device_index()) << kDeviceIndexShift;
  id |= static_cast<int64_t>(task_id.stream_id().device_type()) << kDeviceTypeShift;
  id |= static_cast<int64_t>(task_id.stream_id().rank()) << kRankShift;
  return id;
}

TaskId DecodeTaskIdFromInt64(int64_t task_id_val) {
  int64_t rank = (task_id_val & kRankInt64Mask) >> kRankShift;