diff options
author | 2018-03-14 10:07:34 -0700 | |
---|---|---|
committer | 2018-03-14 10:11:02 -0700 | |
commit | 630247f3b0c6288b17d5a9fceea4567f8d8d799a (patch) | |
tree | 89c227c3d574d578f467e560db578af19aa2d1ec /tensorflow/compiler/xla/service/hlo_module_group_metadata.cc | |
parent | 8a0041122eee68feb34f3579658042d5f62389a7 (diff) |
Add interface in HloModuleGroupMetadata to return all channels
PiperOrigin-RevId: 189044541
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_module_group_metadata.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_module_group_metadata.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc b/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc index eed0112f62..fa5dcb0b36 100644 --- a/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc +++ b/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc @@ -216,6 +216,7 @@ Status HloModuleGroupMetadata::RecordInstructions() { channels_.emplace_back(); channels_.back().id = hlo->channel_id(); channel_id_map_[hlo->channel_id()] = channels_.size() - 1; + max_channel_id_ = std::max(max_channel_id_, hlo->channel_id()); } Channel& channel = channels_[channel_id_map_[hlo->channel_id()]]; |