Function EncodeDeviceIdToInt64

oneflow/core/device/device_id.cpp:31–36

}  // namespace

int64_t EncodeDeviceIdToInt64(const DeviceId& device_id) {
  int64_t id = static_cast<int64_t>(device_id.device_index());
  id |= static_cast<int64_t>(device_id.device_type()) << kDeviceTypeShift;
  id |= static_cast<int64_t>(device_id.rank()) << kRankShift;
  return id;
}

}  // namespace oneflow