OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 00:20:20 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Add more debugging output for XLA send/recv.

Browse Source 
PiperOrigin-RevId: 171041978
This commit is contained in:
A. Unique TensorFlower 2017-10-04 11:58:40 -07:00 committed by TensorFlower Gardener
parent 23992bb091
commit 0578dd65ec
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/compiler/xla/service/channel_tracker.cc
View File

@ -69,7 +69,10 @@ Status ChannelTracker::RegisterSendInternal(const ChannelHandle& handle) {
}
Channel& channel = opaque_to_channel_[handle.handle()];
if (channel.has_sender) {
return FailedPrecondition("channel handle is already used by a sender");
return FailedPrecondition(
"when registering send, passed a channel handle that is already used "
"by a sender: %lld",
handle.handle());
}
channel.has_sender = true;
return Status::OK();
@ -82,7 +85,10 @@ Status ChannelTracker::RegisterRecvInternal(const ChannelHandle& handle) {
Channel& channel = opaque_to_channel_[handle.handle()];
// TODO(b/33942691): Allow more than 1 receivers for broadcast.
if (channel.receiver_count >= 1) {
return FailedPrecondition("channel handle is already used by a receiver");
return FailedPrecondition(
"when registering recv, passed a channel handle that is already used "
"by a receiver: %lld",
handle.handle());
}
channel.receiver_count += 1;
return Status::OK();

4
tensorflow/compiler/xla/service/hlo_instruction.cc
View File

@ -1702,6 +1702,10 @@ std::vector<string> HloInstruction::ExtraAttributesToString() const {
})));
}
if (opcode() == HloOpcode::kSend || opcode() == HloOpcode::kRecv) {
extra.push_back(StrCat("channel_id=", channel_id_));
}
if (opcode() == HloOpcode::kGetTupleElement) {
extra.push_back(StrCat("index=", tuple_index()));
}