message: Change check in EncapsulateClusters to look at assigned device, not requested
diff: @@ -97,20 +97,20 @@ Status EncapsulateClusters(Graph* graph) {
  auto it = device_name_map.find(cluster_idx);
  if (it != device_name_map.end()) {
  - if (it->second != node->requested_device()) {
  + if (it->second != node->assigned_device_name()) {
  std::stringstream ss_err;
  ss_err << "Node " << node->name() << " in cluster " << cluster_idx
  - << " has requested device " << node->requested_device()
  - << " but another node with requested device " << it->second
  + << " has assigned device " << node->assigned_device_name()
  + << " but another node with assigned device " << it->second
  << " has already been seen in the same cluster";
  return errors::Internal(ss_err.str());
  }
  } else {
  NGRAPH_VLOG(3) << "setting cluster " << cluster_idx
  - << " requested device to '" << node->requested_device()
  + << " requested device to '" << node->assigned_device_name()
  << "'";
  - device_name_map[cluster_idx] = node->requested_device();
  + device_name_map[cluster_idx] = node->assigned_device_name();
  }
  }
  